CU-8698jzjj3: fix mypy errors

Author: Tom Searle

medcat-v1/medcat/ner/transformers_ner.py

@@ -177,8 +177,8 @@ def train(self,
               ignore_extra_labels=False,
               dataset=None,
               meta_requirements=None,
-              train_json_path: str=None,
-              test_json_path: str=None,
+              train_json_path: Union[str, list, None]=None,
+              test_json_path: Union[str, list, None]=None,
               trainer_callbacks: Optional[List[Callable[[Trainer], TrainerCallback]]] = None) -> Tuple:
         """Train or continue training a model give a json_path containing a MedCATtrainer export. It will
         continue training if an existing model is loaded or start new training if the model is blank/new.
@@ -571,5 +571,6 @@ def func_has_kwarg(func: Callable, keyword: str):
     test_json_path = 'test_set.json'
     deid_model_path = '/Users/k1897038/Documents/cogstack_docs/medcat_models/medcat_deid_model_691c3f6a6e5400e7.zip'
     deid_cat = DeIdModel.load_model_pack(deid_model_path) 
-    deid_cat.train(train_json_path=train_json_path, test_json_path=test_json_path)
+    deid_cat.train(ignore_extra_labels=True, train_json_path=train_json_path, test_json_path=test_json_path)
+
 

medcat-v1/medcat/utils/ner/deid.py

@@ -69,7 +69,6 @@ def train(self, json_path: Union[str, list, None]=None,
               *args, **kwargs) -> Tuple[Any, Any, Any]:
         assert not all([json_path, train_json_path, test_json_path]), \
                 "Either json_path or train_json_path and test_json_path must be provided when no dataset is provided"
-
         return super().train(json_path=json_path, 
                              train_json_path=train_json_path, 
                              test_json_path=test_json_path, *args, **kwargs)  # type: ignore

medcat-v1/medcat/utils/ner/model.py

@@ -26,6 +26,8 @@ def __init__(self, cat: CAT) -> None:
         self.cat = cat
 
     def train(self, json_path: Union[str, list, None], train_nr: int = 0,
+              train_json_path: Union[str, list, None]=None,
+              test_json_path: Union[str, list, None]=None,
               *args, **kwargs) -> Tuple[Any, Any, Any]:
         """Train the underlying transformers NER model.
 
@@ -40,7 +42,7 @@ def train(self, json_path: Union[str, list, None], train_nr: int = 0,
         Returns:
             Tuple[Any, Any, Any]: df, examples, dataset
         """
-        return self.cat._addl_ner[train_nr].train(json_path, *args, **kwargs)
+        return self.cat._addl_ner[train_nr].train(json_path, train_json_path=train_json_path, test_json_path=test_json_path, *args, **kwargs)
 
     def eval(self, json_path: Union[str, list, None], train_nr: int = 0,
               *args, **kwargs) -> Tuple[Any, Any, Any]: