@@ -37,7 +37,7 @@ def test_finetune_bands_intervals(model_name):
37
37
_ = build_lightning_cli (command_list )
38
38
39
39
@pytest .mark .parametrize ("model_name" , ["prithvi_swin_B" ])
40
- def test_finetune_bands_intervals (model_name ):
40
+ def test_finetune_bands_str (model_name ):
41
41
42
42
model_instance = timm .create_model (model_name )
43
43
@@ -49,22 +49,3 @@ def test_finetune_bands_intervals(model_name):
49
49
command_list = ["fit" , "-c" , f"tests/manufactured-finetune_{ model_name } _string.yaml" ]
50
50
_ = build_lightning_cli (command_list )
51
51
52
-
53
- """
54
- @pytest.mark.parametrize("model_name", ["prithvi_swin_B", "prithvi_swin_L", "prithvi_vit_100", "prithvi_vit_300"])
55
- def test_finetune_multiple_backbones(model_name):
56
-
57
- model_instance = timm.create_model(model_name)
58
- pretrained_bands = [0, 1, 2, 3, 4, 5]
59
- model_bands = [0, 1, 2, 3, 4, 5]
60
-
61
- state_dict = model_instance.state_dict()
62
-
63
- torch.save(state_dict, os.path.join("tests/", model_name + ".pt"))
64
-
65
- # Running the terratorch CLI
66
- command_str = f"python terratorch/__main__.py fit -c tests/manufactured-finetune_{model_name}.yaml"
67
- command_out = subprocess.run(command_str, shell=True)
68
-
69
- assert not command_out.returncode
70
- """
0 commit comments