diff --git a/medcat-v2/medcat/utils/legacy/conversion_all.py b/medcat-v2/medcat/utils/legacy/conversion_all.py index afa37b9d..7d745527 100644 --- a/medcat-v2/medcat/utils/legacy/conversion_all.py +++ b/medcat-v2/medcat/utils/legacy/conversion_all.py @@ -118,7 +118,7 @@ def convert(self) -> CAT: from medcat.utils.legacy.convert_deid import get_trf_ner_from_old trf_ners = [ get_trf_ner_from_old(subfolder, cat.pipe.tokenizer) - for subfolder in os.listdir(self.old_model_folder) + for subfolder in trf_folders ] if len(trf_ners) > 1: raise ValueError("Cannot use more than 1 tranformers NER. "