update extract_skills_config

nestauk · May 8, 2024 · d7bda99 · d7bda99
1 parent 620ea0b
commit d7bda99
Showing 1 changed file with 1 addition and 4 deletions.
diff --git 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: