Skip to content

Commit 44f0597

Browse files
Circumvent boring minor issue for now
Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
1 parent c177911 commit 44f0597

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_finetune.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -51,8 +51,8 @@ def test_finetune_metrics_from_file(model_name):
5151
command_list = ["fit", "-c", f"tests/manufactured-finetune_{model_name}_metrics_from_file.yaml"]
5252
_ = build_lightning_cli(command_list)
5353

54-
def test_finetune_bands_str():
55-
54+
@pytest.mark.parametrize("model_name", ["prithvi_swin_B"])
55+
def test_finetune_bands_str(model_name):
5656
# Running the terratorch CLI
5757
command_list = ["fit", "-c", f"tests/manufactured-finetune_aspphead.yaml"]
5858
_ = build_lightning_cli(command_list)

0 commit comments

Comments
 (0)