Skip to content

Commit

Permalink
ensure type correct
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 987c242 commit 8098584
Showing 1 changed file with 1 addition and 1 deletion.
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 = {i + 1 for i in range(132)} - {16, 18, 129, 130, 131}
supported_list = list({i + 1 for i in range(132)} - {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 8098584

Please sign in to comment.