Skip to content

Commit 10f3386

Browse files
This test is no longer necessary
Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
1 parent a02b6cd commit 10f3386

File tree

1 file changed

+0
-10
lines changed

1 file changed

+0
-10
lines changed

tests/test_backbones.py

-10
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import pytest
22
import timm
33
import torch
4-
import importlib
54

65
import terratorch # noqa: F401
76

@@ -50,16 +49,7 @@ def test_vit_models_accept_multitemporal(model_name, input_224_multitemporal):
5049
backbone = timm.create_model(model_name, pretrained=False, num_frames=NUM_FRAMES)
5150
backbone(input_224_multitemporal)
5251

53-
# Swin IS NOT on HuggingFace
54-
@pytest.mark.parametrize("model_name", ["prithvi_swin_L", "prithvi_swin_B"])
55-
def test_swin_instantiation(model_name):
56-
base_module = "terratorch.models.backbones.prithvi_swin"
57-
module = importlib.import_module(base_module)
58-
model_class = getattr(module, model_name)
5952

60-
model = model_class(pretrained=False, pretrained_bands=[0,1,2,3,4,5,6,7,8,9],
61-
bands=[1,2,3,4,5,6])
62-
6353
#def test_swin_models_accept_non_divisible_by_patch_size(input_386):
6454
# backbone = timm.create_model("prithvi_swin_90_us", pretrained=False, num_frames=NUM_FRAMES)
6555
# backbone(input_386)

0 commit comments

Comments
 (0)