Skip to content

Commit

Permalink
fix code format
Browse files Browse the repository at this point in the history
Signed-off-by: Yiheng Wang <vennw@nvidia.com>
  • Loading branch information
yiheng-wang-nv committed Oct 10, 2024
1 parent d6569e6 commit 67cb148
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion ci/unit_tests/test_vista3d.py
Original file line number Diff line number Diff line change
Expand Up @@ -242,7 +242,7 @@
"initialize": ["$monai.utils.set_determinism(seed=123)"],
"error": "Undefined label prompt detected. Provide point prompts for zero-shot.",
}
]
],
]


Expand Down
2 changes: 1 addition & 1 deletion models/vista3d/scripts/evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -167,7 +167,7 @@ def check_prompts_format(self, label_prompt, points, point_labels):
if not np.all([(x < 255).item() for x in label_prompt]):
raise ValueError("Current bundle only supports label prompt smaller than 255.")
if points is None:
supported_list = list(set([i+1 for i in range(132)]) - set([16,18,129,130,131]))
supported_list = list(set([i + 1 for i in range(132)]) - set([16, 18, 129, 130, 131]))
if not np.all([x in supported_list for x in label_prompt]):
raise ValueError("Undefined label prompt detected. Provide point prompts for zero-shot.")
else:
Expand Down

0 comments on commit 67cb148

Please sign in to comment.