remove swin backbone from vit-only tests

Signed-off-by: CarlosGomes98 <[email protected]>
IBM · Jun 28, 2024 · 313b9f9 · 313b9f9
1 parent c2064f6
commit 313b9f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_backbones.py b/tests/test_backbones.py
@@ -44,7 +44,7 @@ def test_can_create_backbones_from_timm_features_only(model_name, test_input, re
     backbone(input_tensor)
 
 
-@pytest.mark.parametrize("model_name", ["prithvi_vit_100", "prithvi_vit_300", "prithvi_swin_B"])
+@pytest.mark.parametrize("model_name", ["prithvi_vit_100", "prithvi_vit_300"])
 def test_vit_models_accept_multitemporal(model_name, input_224_multitemporal):
     backbone = timm.create_model(model_name, pretrained=False, num_frames=NUM_FRAMES)
     backbone(input_224_multitemporal)