Update test_pytorch_models.py

MMathisLab · web-flow · commit ff25a26bcff9 · 2025-07-05T09:02:46.000+02:00
diff --git a/tests/test_pytorch_models.py b/tests/test_pytorch_models.py
@@ -21,6 +21,7 @@
         ("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]]):

Original file line number	Diff line number	Diff line change
`@@ -21,6 +21,7 @@`
`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"]),`
`24`	`25`	`]`
`25`	`26`	`)`
`26`	`27`	`def test_get_super_animal_detectors(data: tuple[str, list[str]]):`