CU-8694pey4u: add type hints

CogStack · May 30, 2024 · 661ab13 · 661ab13
1 parent f836f65
commit 661ab13
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/medcat/cat.py b/medcat/cat.py
@@ -390,7 +390,7 @@ def load_model_pack(cls,
         model_pack_path = cls.attempt_unpack(zip_path)
 
         # Load the CDB
-        cdb = cls.load_cdb(model_pack_path)
+        cdb: CDB = cls.load_cdb(model_pack_path)
 
         # load config
         config_path = os.path.join(model_pack_path, "config.json")
@@ -419,7 +419,7 @@ def load_model_pack(cls,
         # Find metacat models in the model_pack
         meta_cats = []
         if load_meta_models:
-            meta_cats = [mc[1] for mc in cls.load_meta_cats(model_pack_path)]
+            meta_cats: List[MetaCAT] = [mc[1] for mc in cls.load_meta_cats(model_pack_path)]
 
         # Find Rel models in model_pack
         rel_paths = [os.path.join(model_pack_path, path) for path in os.listdir(model_pack_path) if path.startswith('rel_')] if load_rel_models else []