We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 227138d commit 9ebcee1Copy full SHA for 9ebcee1
tests/nn/test_training.py
@@ -389,7 +389,7 @@ def test_resume_training_cli(
389
assert trainer.config.model.base_checkpoint == base_checkpoint_path
390
391
392
-@pytest.mark.parametrize("keep_viz_cli", [None, "--keep_viz"])
+@pytest.mark.parametrize("keep_viz_cli", ["", "--keep_viz"])
393
def test_keep_viz_cli(
394
keep_viz_cli,
395
min_single_instance_robot_model_path: str,
@@ -407,4 +407,6 @@ def test_keep_viz_cli(
407
trainer = sleap_train(cli_args)
408
409
# Check that --keep_viz is set correctly
410
- assert trainer.config.outputs.keep_viz_images == bool(keep_viz_cli)
+ assert trainer.config.outputs.keep_viz_images == (
411
+ True if keep_viz_cli == "--keep_viz" else False
412
+ )
0 commit comments