update zero shot test case

Signed-off-by: Yiheng Wang <vennw@nvidia.com>
Project-MONAI · Oct 10, 2024 · 1e33e1e · 1e33e1e
1 parent 8098584
commit 1e33e1e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ci/unit_tests/test_vista3d.py b/ci/unit_tests/test_vista3d.py
@@ -61,7 +61,7 @@
 TEST_CASE_INFER_MULTI_NEW_STR_PROMPT = [
     {
         "bundle_root": "models/vista3d",
-        "input_dict": {"label_prompt": ["new class 1", "new class 2", "new class 3"]},
+        "input_dict": {"label_prompt": ["new class 1"], "points": [[123, 212, 151]], "point_labels": [1]},
         "patch_size": [32, 32, 32],
         "checkpointloader#_disabled_": True,  # do not load weights"
         "initialize": ["$monai.utils.set_determinism(seed=123)"],