fix test predict --example command

FynnBe · FynnBe · commit 5f52b15e0960 · 2024-11-12T09:34:14.000+01:00
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -35,7 +35,7 @@ def run_subprocess(
             "pytorch_state_dict",
         ],
         ["test", "unet2d_nuclei_broad_model"],
-        ["predict", "--example"],
+        ["predict", "--example", "unet2d_nuclei_broad_model"],
     ],
 )
 def test_cli(args: List[str], unet2d_nuclei_broad_model: str):

Original file line number	Diff line number	Diff line change
`@@ -35,7 +35,7 @@ def run_subprocess(`
`35`	`35`	`"pytorch_state_dict",`
`36`	`36`	`],`
`37`	`37`	`["test", "unet2d_nuclei_broad_model"],`
`38`		`- ["predict", "--example"],`
	`38`	`+ ["predict", "--example", "unet2d_nuclei_broad_model"],`
`39`	`39`	`],`
`40`	`40`	`)`
`41`	`41`	`def test_cli(args: List[str], unet2d_nuclei_broad_model: str):`