Fixed type check.

CogStack · Feb 19, 2024 · bb53396 · bb53396
1 parent f00a74a
commit bb53396
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/medcat_service/nlp_processor/medcat_processor.py b/medcat_service/nlp_processor/medcat_processor.py
@@ -219,9 +219,9 @@ def retrain_medcat(self, content, replace_cdb):
 
     def _populate_model_card_info(self, config: Config):
         self.model_card_info["ontologies"] = config.version.ontology \
-            if (type(config.version.ontology) == list) else str(config.version.ontology)
+            if (isinstance(config.version.ontology, list)) else str(config.version.ontology)
         self.model_card_info["meta_cat_model_names"] = [i["Category Name"] for i in config.version.meta_cats] \
-            if (type(config.version.meta_cats) == list) else str(config.version.meta_cats)
+            if (isinstance(config.version.meta_cats, list)) else str(config.version.meta_cats)
         self.model_card_info["model_last_modified_on"] = str(config.version.last_modified)
 
     # helper MedCAT methods