No ViT tests

Author: adamjstewart

pyproject.toml

@@ -291,7 +291,7 @@ testpaths = [
     "tests",
     "docs/tutorials",
 ]
-tmp_path_retention_policy = "none"
+tmp_path_retention_policy = "failed"
 
 # https://docs.astral.sh/ruff/settings/
 [tool.ruff]

tests/models/test_vit.py

@@ -70,9 +70,9 @@ def test_transforms(self, weights: WeightsEnum) -> None:
         }
         weights.transforms(sample)
 
-    @pytest.mark.slow
-    def test_vit_download(self, weights: WeightsEnum) -> None:
-        vit_small_patch16_224(weights=weights)
+    # @pytest.mark.slow
+    # def test_vit_download(self, weights: WeightsEnum) -> None:
+    #     vit_small_patch16_224(weights=weights)
 
 
 class TestViTBase16:
@@ -119,9 +119,9 @@ def test_transforms(self, weights: WeightsEnum) -> None:
         }
         weights.transforms(sample)
 
-    @pytest.mark.slow
-    def test_vit_download(self, weights: WeightsEnum) -> None:
-        vit_base_patch16_224(weights=weights)
+    # @pytest.mark.slow
+    # def test_vit_download(self, weights: WeightsEnum) -> None:
+    #     vit_base_patch16_224(weights=weights)
 
 
 class TestViTLarge16:
@@ -274,9 +274,9 @@ def test_transforms(self, weights: WeightsEnum) -> None:
         sample = {'image': torch.arange(c * h * w, dtype=torch.float).view(c, h, w)}
         weights.transforms(sample)
 
-    @pytest.mark.slow
-    def test_vit_download(self, weights: WeightsEnum) -> None:
-        vit_small_patch14_dinov2(weights=weights)
+    # @pytest.mark.slow
+    # def test_vit_download(self, weights: WeightsEnum) -> None:
+    #     vit_small_patch14_dinov2(weights=weights)
 
 
 class TestViTBase14_DINOv2:
@@ -327,6 +327,6 @@ def test_transforms(self, weights: WeightsEnum) -> None:
         sample = {'image': torch.arange(c * h * w, dtype=torch.float).view(c, h, w)}
         weights.transforms(sample)
 
-    @pytest.mark.slow
-    def test_vit_download(self, weights: WeightsEnum) -> None:
-        vit_base_patch14_dinov2(weights=weights)
+    # @pytest.mark.slow
+    # def test_vit_download(self, weights: WeightsEnum) -> None:
+    #     vit_base_patch14_dinov2(weights=weights)