Update test_pytorch_models.py

AlexEMG · web-flow · commit 6e1e39fd238b · 2025-07-05T09:58:42.000+02:00
diff --git a/tests/test_pytorch_models.py b/tests/test_pytorch_models.py
@@ -16,12 +16,11 @@
 
 
 @pytest.mark.parametrize(
-    "data",
+    "detector_data",
     [
         ("superanimal_bird", ["ssdlite"]),
         ("superanimal_topviewmouse", ["fasterrcnn_resnet50_fpn_v2"]),
         ("superanimal_quadruped", ["fasterrcnn_resnet50_fpn_v2"]),
-        ("superanimal_humanbody", ["rtmpose_x"]),
     ]
 )
 def test_get_super_animal_detectors(data: tuple[str, list[str]]):
@@ -33,11 +32,12 @@ def test_get_super_animal_detectors(data: tuple[str, list[str]]):
 
 
 @pytest.mark.parametrize(
-    "data",
+    "posemodel_data",
     [
         ("superanimal_bird", ["resnet_50"]),
         ("superanimal_topviewmouse", ["hrnet_w32"]),
         ("superanimal_quadruped", ["hrnet_w32"]),
+        ("superanimal_humanbody", ["rtmpose_x"]),
     ]
 )
 def test_get_super_animal_pose_models(data: tuple[str, list[str]]):

Original file line number	Diff line number	Diff line change
`@@ -16,12 +16,11 @@`
`16`	`16`
`17`	`17`
`18`	`18`	`@pytest.mark.parametrize(`
`19`		`- "data",`
	`19`	`+ "detector_data",`
`20`	`20`	`[`
`21`	`21`	`("superanimal_bird", ["ssdlite"]),`
`22`	`22`	`("superanimal_topviewmouse", ["fasterrcnn_resnet50_fpn_v2"]),`
`23`	`23`	`("superanimal_quadruped", ["fasterrcnn_resnet50_fpn_v2"]),`
`24`		`- ("superanimal_humanbody", ["rtmpose_x"]),`
`25`	`24`	`]`
`26`	`25`	`)`
`27`	`26`	`def test_get_super_animal_detectors(data: tuple[str, list[str]]):`
`@@ -33,11 +32,12 @@ def test_get_super_animal_detectors(data: tuple[str, list[str]]):`
`33`	`32`
`34`	`33`
`35`	`34`	`@pytest.mark.parametrize(`
`36`		`- "data",`
	`35`	`+ "posemodel_data",`
`37`	`36`	`[`
`38`	`37`	`("superanimal_bird", ["resnet_50"]),`
`39`	`38`	`("superanimal_topviewmouse", ["hrnet_w32"]),`
`40`	`39`	`("superanimal_quadruped", ["hrnet_w32"]),`
	`40`	`+ ("superanimal_humanbody", ["rtmpose_x"]),`
`41`	`41`	`]`
`42`	`42`	`)`
`43`	`43`	`def test_get_super_animal_pose_models(data: tuple[str, list[str]]):`