diff --git a/medcat/cat.py b/medcat/cat.py index 2323cd737..1e6736a36 100644 --- a/medcat/cat.py +++ b/medcat/cat.py @@ -1332,7 +1332,7 @@ def multiprocessing(self, nproc: int = 2, batch_size_chars: int = 5000 * 1000, only_cui: bool = False, - addl_info: List[str] = [], + addl_info: List[str] = ['cui2icd10', 'cui2ontologies', 'cui2snomed'], separate_nn_components: bool = True, out_split_size_chars: Optional[int] = None, save_dir_path: str = os.path.abspath(os.getcwd()), @@ -1528,7 +1528,7 @@ def multiprocessing_pipe(self, nproc: Optional[int] = None, batch_size: Optional[int] = None, only_cui: bool = False, - addl_info: List[str] = [], + addl_info: List[str] = ['cui2icd10', 'cui2ontologies', 'cui2snomed'], return_dict: bool = True, batch_factor: int = 2) -> Union[List[Tuple], Dict]: """Run multiprocessing NOT FOR TRAINING