Merge branch 'anipose-loader' of https://github.com/neuroinformatics-…

…unit/movement into anipose-loader # Conflicts: # movement/io/load_poses.py # tests/test_unit/test_load_poses.py
neuroinformatics-unit · Dec 10, 2024 · 481e80b · 481e80b
2 parents 100008c + ead27dc
commit 481e80b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 11 deletions.
diff --git a/movement/io/load_poses.py b/movement/io/load_poses.py
@@ -96,10 +96,9 @@ def from_numpy(
 
 def from_file(
     file_path: Path | str,
-    source_software: Literal["DeepLabCut",
-                             "SLEAP",
-                             "LightningPose",
-                             "Anipose"] | None = None,
+    source_software: Literal[
+        "DeepLabCut", "SLEAP", "LightningPose", "Anipose"
+    ],
     fps: float | None = None,
     **kwargs,
 ) -> xr.Dataset:

diff --git a/tests/test_unit/test_load_poses.py b/tests/test_unit/test_load_poses.py
@@ -258,13 +258,7 @@ def test_load_multi_individual_from_lp_file_raises():
 
 @pytest.mark.parametrize(
     "source_software",
-    [
-        "SLEAP",
-        "DeepLabCut",
-        "LightningPose",
-        "Anipose",
-        "Unknown",
-    ],
+    ["SLEAP", "DeepLabCut", "LightningPose", "Anipose", "Unknown"],
 )
 @pytest.mark.parametrize("fps", [None, 30, 60.0])
 def test_from_file_delegates_correctly(source_software, fps):