Set the cli back

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

Author: Joao-L-S-Almeida

tests/test_finetune.py

@@ -7,7 +7,7 @@
 import os 
 
 from terratorch.cli_tools import build_lightning_cli
-"""
+
 @pytest.mark.parametrize("model_name", ["prithvi_swin_B", "prithvi_swin_L", "prithvi_vit_100", "prithvi_vit_300"])
 def test_finetune_multiple_backbones(model_name):
 
@@ -22,8 +22,8 @@ def test_finetune_multiple_backbones(model_name):
     # Running the terratorch CLI
     command_list = ["fit", "-c", f"tests/manufactured-finetune_{model_name}.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):
 
@@ -40,4 +40,4 @@ def test_finetune_multiple_backbones(model_name):
     command_out = subprocess.run(command_str, shell=True)
     
     assert not command_out.returncode
-    
+ """