From 6c7842323a2df342468f9da1e14503b62360a6f1 Mon Sep 17 00:00:00 2001 From: Yiheng Wang Date: Thu, 10 Oct 2024 22:37:48 +0800 Subject: [PATCH] remove unnecessary list Signed-off-by: Yiheng Wang --- models/vista3d/scripts/evaluator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/vista3d/scripts/evaluator.py b/models/vista3d/scripts/evaluator.py index a95bdf30..3f0b7777 100644 --- a/models/vista3d/scripts/evaluator.py +++ b/models/vista3d/scripts/evaluator.py @@ -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 = set(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: