Update utils.py

Author: C-Achard

dlclive/modelzoo/utils.py

@@ -31,7 +31,7 @@ def get_super_animal_project_config_path(super_animal: str) -> Path:
     cfg_path = _MODELZOO_PATH / "project_configs" / f"{super_animal}.yaml"
     if not cfg_path.exists():
         raise FileNotFoundError(
-            f"Modelzoo project configuration file not found: {cfg_path}Available projects: {list_available_projects()}"
+            f"Modelzoo project configuration file not found: {cfg_path} Available projects: {list_available_projects()}"
         )
     return cfg_path
 
@@ -89,7 +89,8 @@ def add_metadata(
     config["metadata"] = {
         "project_path": project_config["project_path"],
         "pose_config_path": "",
-        "bodyparts": project_config.get("multianimalbodyparts") or project_config["bodyparts"],
+        "bodyparts": project_config.get("multianimalbodyparts")
+        or project_config["bodyparts"],
         "unique_bodyparts": project_config.get("uniquebodyparts", []),
         "individuals": project_config.get("individuals", ["animal"]),
         "with_identity": project_config.get("identity", False),
@@ -131,7 +132,9 @@ def load_super_animal_config(
     else:
         model_config["method"] = "TD"
         if super_animal != "superanimal_humanbody":
-            detector_cfg_path = get_super_animal_model_config_path(model_name=detector_name)
+            detector_cfg_path = get_super_animal_model_config_path(
+                model_name=detector_name
+            )
             detector_cfg = read_config_as_dict(detector_cfg_path)
             model_config["detector"] = detector_cfg
     return model_config
@@ -160,13 +163,17 @@ def download_super_animal_snapshot(dataset: str, model_name: str) -> Path:
         return model_path
 
     try:
-        download_huggingface_model(model_name, target_dir=str(snapshot_dir), rename_mapping=model_filename)
+        download_huggingface_model(
+            model_name, target_dir=str(snapshot_dir), rename_mapping=model_filename
+        )
 
         if not model_path.exists():
             raise RuntimeError(f"Failed to download {model_name} to {model_path}")
 
     except Exception as e:
-        logging.error(f"Failed to download superanimal snapshot {model_name} to {model_path}: {e}")
+        logging.error(
+            f"Failed to download superanimal snapshot {model_name} to {model_path}: {e}"
+        )
         raise e
 
     return model_path