Skip to content

Commit

Permalink
Set the cli back
Browse files Browse the repository at this point in the history
Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
  • Loading branch information
Joao-L-S-Almeida committed Jul 3, 2024
1 parent d8b0263 commit 168fe90
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/test_finetune.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):

Expand All @@ -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):
Expand All @@ -40,4 +40,4 @@ def test_finetune_multiple_backbones(model_name):
command_out = subprocess.run(command_str, shell=True)
assert not command_out.returncode

"""

0 comments on commit 168fe90

Please sign in to comment.