diff --git a/ojd_daps_skills/extract_skills/extract_skills_utils.py b/ojd_daps_skills/extract_skills/extract_skills_utils.py index ddb7298d..0be4212a 100644 --- a/ojd_daps_skills/extract_skills/extract_skills_utils.py +++ b/ojd_daps_skills/extract_skills/extract_skills_utils.py @@ -75,7 +75,7 @@ class Config: @classmethod def create( - cls, ner_model_name: Optional[str] = None, ms_model_name: Optional[str] = None + cls, ner_model_name: str, ms_model_name: str ) -> "ExtractConfig": """ Creates an instance of ExtractConfig by loading configurations. @@ -94,9 +94,6 @@ def create( """ # set Doc extension here # Use default values if none provided - ner_model_name = ner_model_name or cls.ner_model_name - ms_model_name = ms_model_name or cls.ms_model_name - Doc.set_extension("skill_spans", default=[], force=True) if "/" in ner_model_name: