Testing to use strings to define a model

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>

Author: Joao-L-S-Almeida

terratorch/datasets/generic_pixel_wise_dataset.py

@@ -135,7 +135,7 @@ def __init__(
             bands_type = self._bands_as_int_or_str(dataset_bands, output_bands)
 
             if bands_type == str:
-                raise UserWarning("When the bands are defined as str, guarantee your input files"+ 
+                UserWarning("When the bands are defined as str, guarantee your input files"+ 
                                   "are organized by band and all have its specific name.")
 
         if self.output_bands and not self.dataset_bands:
@@ -203,7 +203,7 @@ def _bands_as_int_or_str(self, dataset_bands, output_bands) -> type:
                 band_type[b] = str
             else:
                 pass 
-        if band_type.cound(band_type[0]) == len(band_type):
+        if band_type.count(band_type[0]) == len(band_type):
             return band_type[0]
         else:
             raise Exception("The bands must be or all str or all int.")

tests/test_finetune.py

@@ -36,6 +36,20 @@ def test_finetune_bands_intervals(model_name):
     command_list = ["fit", "-c", f"tests/manufactured-finetune_{model_name}_band_interval.yaml"]
     _ = build_lightning_cli(command_list)
 
+@pytest.mark.parametrize("model_name", ["prithvi_swin_B"])
+def test_finetune_bands_intervals(model_name):
+
+    model_instance = timm.create_model(model_name)
+
+    state_dict = model_instance.state_dict()
+
+    torch.save(state_dict, os.path.join("tests/", model_name + ".pt"))
+
+    # Running the terratorch CLI
+    command_list = ["fit", "-c", f"tests/manufactured-finetune_{model_name}_string.yaml"]
+    _ = build_lightning_cli(command_list)
+
+
 """
 @pytest.mark.parametrize("model_name", ["prithvi_swin_B", "prithvi_swin_L", "prithvi_vit_100", "prithvi_vit_300"])
 def test_finetune_multiple_backbones(model_name):