Update code duplication marker

Author: C-Achard

dlclive/core/inferenceutils.py

@@ -10,7 +10,8 @@
 #
 
 
-# NOTE DUPLICATED from deeplabcut/core/inferenceutils.py
+# NOTE - DUPLICATED @C-Achard 2026-26-01: Copied from the original DeepLabCut codebase
+# from deeplabcut/core/inferenceutils.py
 from __future__ import annotations
 
 import heapq

dlclive/modelzoo/resolve_config.py

@@ -1,9 +1,10 @@
 """
-Helper function to deal with default values in the model configuration. 
+Helper function to deal with default values in the model configuration.
 For instance, "num_bodyparts x 2" is replaced with the number of bodyparts multiplied by 2.
 """
-# NOTE JR 2026-23-01: This is duplicate code, copied from the original DeepLabCut-Live codebase.
 
+# NOTE - DUPLICATED @deruyter92 2026-23-01: Copied from the original DeepLabCut codebase
+# from deeplabcut/pose_estimation_pytorch/modelzoo/utils.py
 import copy
 
 
@@ -78,8 +79,7 @@ def get_updated_value(variable: str) -> int | list[int]:
         var_name = var_parts[0]
         if updated_values[var_name] is None:
             raise ValueError(
-                f"Found {variable} in the configuration file, but there is no default "
-                f"value for this variable."
+                f"Found {variable} in the configuration file, but there is no default value for this variable."
             )
 
         if len(var_parts) == 1:
@@ -99,9 +99,7 @@ def get_updated_value(variable: str) -> int | list[int]:
             else:
                 raise ValueError(f"Unknown operator for variable: {variable}")
 
-        raise ValueError(
-            f"Found {variable} in the configuration file, but cannot parse it."
-        )
+        raise ValueError(f"Found {variable} in the configuration file, but cannot parse it.")
 
     updated_values = {
         "num_bodyparts": num_bodyparts,
@@ -127,10 +125,7 @@ def get_updated_value(variable: str) -> int | list[int]:
                 backbone_output_channels,
                 **kwargs,
             )
-        elif (
-            isinstance(config[k], str)
-            and config[k].strip().split(" ")[0] in updated_values.keys()
-        ):
+        elif isinstance(config[k], str) and config[k].strip().split(" ")[0] in updated_values.keys():
             config[k] = get_updated_value(config[k])
 
-    return config
+    return config

dlclive/modelzoo/utils.py

@@ -5,55 +5,53 @@
 # This should be removed once a solution is found to address duplicate code.
 
 import copy
-from pathlib import Path
 import logging
+from pathlib import Path
 
+from dlclibrary.dlcmodelzoo.modelzoo_download import download_huggingface_model
 from ruamel.yaml import YAML
 
-from dlclibrary.dlcmodelzoo.modelzoo_download import download_huggingface_model
 from dlclive.modelzoo.resolve_config import update_config
 
-_MODELZOO_PATH = Path(__file__).parent 
+_MODELZOO_PATH = Path(__file__).parent
 
 
 def get_super_animal_model_config_path(model_name: str) -> Path:
     """Get the path to the model configuration file for a model and validate choice of model"""
-    cfg_path = _MODELZOO_PATH / 'model_configs' / f"{model_name}.yaml"
+    cfg_path = _MODELZOO_PATH / "model_configs" / f"{model_name}.yaml"
     if not cfg_path.exists():
         raise FileNotFoundError(
-            f"Modelzoo model configuration file not found: {cfg_path} "
-            f"Available models: {list_available_models()}"
+            f"Modelzoo model configuration file not found: {cfg_path} Available models: {list_available_models()}"
         )
     return cfg_path
 
 
 def get_super_animal_project_config_path(super_animal: str) -> Path:
     """Get the path to the project configuration file for a project and validate choice of project"""
-    cfg_path = _MODELZOO_PATH / 'project_configs' / f"{super_animal}.yaml"
+    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}"
-            f"Available projects: {list_available_projects()}"
+            f"Modelzoo project configuration file not found: {cfg_path}Available projects: {list_available_projects()}"
         )
     return cfg_path
 
 
 def get_snapshot_folder_path() -> Path:
-    return _MODELZOO_PATH / 'snapshots'
+    return _MODELZOO_PATH / "snapshots"
 
 
 def list_available_models() -> list[str]:
-    return [p.stem for p in _MODELZOO_PATH.glob('model_configs/*.yaml')]
+    return [p.stem for p in _MODELZOO_PATH.glob("model_configs/*.yaml")]
 
 
 def list_available_projects() -> list[str]:
-    return [p.stem for p in _MODELZOO_PATH.glob('project_configs/*.yaml')]
+    return [p.stem for p in _MODELZOO_PATH.glob("project_configs/*.yaml")]
 
 
 def list_available_combinations() -> list[str]:
     models = list_available_models()
     projects = list_available_projects()
-    combinations = ['_'.join([p, m]) for p in projects for m in models]
+    combinations = ["_".join([p, m]) for p in projects for m in models]
     return combinations
 
 
@@ -65,14 +63,18 @@ def read_config_as_dict(config_path: str | Path) -> dict:
     Returns:
         The configuration file with pure Python classes
     """
-    with open(config_path, "r") as f:
-        cfg = YAML(typ='safe', pure=True).load(f)
+    with open(config_path) as f:
+        cfg = YAML(typ="safe", pure=True).load(f)
 
     return cfg
 
 
-# NOTE JR 2026-23-01: This is duplicate code, copied from the original DeepLabCut-Live codebase.
-def add_metadata(project_config: dict, config: dict,) -> dict:
+# NOTE - DUPLICATED @deruyter92 2026-23-01: Copied from the original DeepLabCut codebase
+# from deeplabcut/pose_estimation_pytorch/config/make_pose_config.py
+def add_metadata(
+    project_config: dict,
+    config: dict,
+) -> dict:
     """Adds metadata to a pytorch pose configuration
 
     Args:
@@ -95,7 +97,8 @@ def add_metadata(project_config: dict, config: dict,) -> dict:
     return config
 
 
-# NOTE JR 2026-23-01: This is duplicate code, copied from the original DeepLabCut-Live codebase.
+# NOTE - DUPLICATED @deruyter92 2026-23-01: Copied from the original DeepLabCut codebase
+# from deeplabcut/pose_estimation_pytorch/modelzoo/utils.py
 def load_super_animal_config(
     super_animal: str,
     model_name: str,
@@ -128,9 +131,7 @@ 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
@@ -159,9 +160,7 @@ 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}")
@@ -171,5 +170,3 @@ def download_super_animal_snapshot(dataset: str, model_name: str) -> Path:
         raise e
 
     return model_path
-
-

pyproject.toml

@@ -97,5 +97,5 @@ target-version = "py310"
 fix = true
 line-length = 120
 
-[tool.ruff.pydocstyle]
+[tool.ruff.lint.pydocstyle]
 convention = "google"

tests/test_modelzoo.py

@@ -1,17 +1,15 @@
-# NOTE JR 2026-23-01: This is duplicate code, copied from the original DeepLabCut-Live codebase.
-
+# NOTE - DUPLICATED @deruyter92 2026-23-01: Copied from the original DeepLabCut codebase
+# from deeplabcut/tests/pose_estimation_pytorch/modelzoo/test_modelzoo_utils.py
 import os
 
-import pytest
 import dlclibrary
+import pytest
 from dlclibrary.dlcmodelzoo.modelzoo_download import MODELOPTIONS
 
 from dlclive import modelzoo
 
 
-@pytest.mark.parametrize(
-    "super_animal", ["superanimal_quadruped", "superanimal_topviewmouse"]
-)
+@pytest.mark.parametrize("super_animal", ["superanimal_quadruped", "superanimal_topviewmouse"])
 @pytest.mark.parametrize("model_name", ["hrnet_w32"])
 @pytest.mark.parametrize("detector_name", [None, "fasterrcnn_resnet50_fpn_v2"])
 def test_get_config_model_paths(super_animal, model_name, detector_name):
@@ -48,4 +46,4 @@ def test_download_huggingface_wrong_model():
 @pytest.mark.skip(reason="slow")
 @pytest.mark.parametrize("model", MODELOPTIONS)
 def test_download_all_models(tmp_path_factory, model):
-    test_download_huggingface_model(tmp_path_factory, model)
+    test_download_huggingface_model(tmp_path_factory, model)