ensure type correct

Signed-off-by: Yiheng Wang <[email protected]>

Author: yiheng-wang-nv

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 = {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: