Skip to content

Commit 313b9f9

Browse files
committed
remove swin backbone from vit-only tests
Signed-off-by: CarlosGomes98 <[email protected]>
1 parent c2064f6 commit 313b9f9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_backbones.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def test_can_create_backbones_from_timm_features_only(model_name, test_input, re
4444
backbone(input_tensor)
4545

4646

47-
@pytest.mark.parametrize("model_name", ["prithvi_vit_100", "prithvi_vit_300", "prithvi_swin_B"])
47+
@pytest.mark.parametrize("model_name", ["prithvi_vit_100", "prithvi_vit_300"])
4848
def test_vit_models_accept_multitemporal(model_name, input_224_multitemporal):
4949
backbone = timm.create_model(model_name, pretrained=False, num_frames=NUM_FRAMES)
5050
backbone(input_224_multitemporal)

0 commit comments

Comments
 (0)