Skip to content

Commit 7a94a82

Browse files
This test is no longer required
Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
1 parent de533dd commit 7a94a82

File tree

1 file changed

+1
-20
lines changed

1 file changed

+1
-20
lines changed

tests/test_finetune.py

+1-20
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def test_finetune_bands_intervals(model_name):
3737
_ = build_lightning_cli(command_list)
3838

3939
@pytest.mark.parametrize("model_name", ["prithvi_swin_B"])
40-
def test_finetune_bands_intervals(model_name):
40+
def test_finetune_bands_str(model_name):
4141

4242
model_instance = timm.create_model(model_name)
4343

@@ -49,22 +49,3 @@ def test_finetune_bands_intervals(model_name):
4949
command_list = ["fit", "-c", f"tests/manufactured-finetune_{model_name}_string.yaml"]
5050
_ = build_lightning_cli(command_list)
5151

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

Comments
 (0)