Workaround for osmr/imgclsmob#103

lrzpellegrini · Nov 8, 2022 · c6997af · c6997af
1 parent e6bcd94
commit c6997af
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_models.py b/tests/test_models.py
@@ -82,9 +82,9 @@ def test_densenet(self):
 
     def test_get_model(self):
         # Check general wrapper and whether downloading pretrained model works
-        model = get_model("simplepose_resnet18_coco", pretrained=True)
+        model = get_model("resnet10", pretrained=True)
         self.assertIsInstance(
-            model, pytorchcv.models.simplepose_coco.SimplePose
+            model, pytorchcv.models.resnet.ResNet
         )