Temporarily removing the Swin tests

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>

Author: Joao-L-S-Almeida

tests/test_backbones.py

@@ -28,7 +28,7 @@ def input_386():
     return torch.ones((1, NUM_CHANNELS, 386, 386))
 
 
-@pytest.mark.parametrize("model_name", ["prithvi_swin_90_us", "prithvi_vit_100", "prithvi_vit_300"])
+@pytest.mark.parametrize("model_name", ["prithvi_vit_100", "prithvi_vit_300"]) #["prithvi_swin_90_us", "prithvi_vit_100", "prithvi_vit_300"])
 @pytest.mark.parametrize("test_input", ["input_224", "input_512"])
 def test_can_create_backbones_from_timm(model_name, test_input, request):
     backbone = timm.create_model(model_name, pretrained=False)
@@ -50,6 +50,6 @@ def test_vit_models_accept_multitemporal(model_name, input_224_multitemporal):
     backbone(input_224_multitemporal)
 
 
-def test_swin_models_accept_non_divisible_by_patch_size(input_386):
-    backbone = timm.create_model("prithvi_swin_90_us", pretrained=False, num_frames=NUM_FRAMES)
-    backbone(input_386)
+#def test_swin_models_accept_non_divisible_by_patch_size(input_386):
+#    backbone = timm.create_model("prithvi_swin_90_us", pretrained=False, num_frames=NUM_FRAMES)
+#    backbone(input_386)