Circumvent boring minor issue for now

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Aug 9, 2024 · 44f0597 · 44f0597
1 parent c177911
commit 44f0597
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_finetune.py b/tests/test_finetune.py
@@ -51,8 +51,8 @@ def test_finetune_metrics_from_file(model_name):
     command_list = ["fit", "-c", f"tests/manufactured-finetune_{model_name}_metrics_from_file.yaml"]
     _ = build_lightning_cli(command_list)
 
-def test_finetune_bands_str():
-
+@pytest.mark.parametrize("model_name", ["prithvi_swin_B"])
+def test_finetune_bands_str(model_name):
     # Running the terratorch CLI
     command_list = ["fit", "-c", f"tests/manufactured-finetune_aspphead.yaml"]
     _ = build_lightning_cli(command_list)