fix test

Lightning-AI · Mar 30, 2024 · bacd1d6 · bacd1d6
1 parent 296101d
commit bacd1d6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_evaluate.py b/tests/test_evaluate.py
@@ -54,6 +54,7 @@ def test_evaluate_script(tmp_path, fake_checkpoint_dir, monkeypatch):
         module.convert_and_evaluate(**fn_kwargs)
 
 
+@pytest.mark.parametrize("mode", ["file", "entrypoint"])
 def test_cli():
     cli_path = Path(__file__).parent.parent / "eval" / "evaluate.py"
     output = subprocess.check_output([sys.executable, cli_path, f"--checkpoint_dir {fake_checkpoint_dir}", "-h"])