From 67cb1485915ffe014667d919435454aebf73f1fe Mon Sep 17 00:00:00 2001
From: Yiheng Wang <vennw@nvidia.com>
Date: Thu, 10 Oct 2024 22:33:31 +0800
Subject: [PATCH] fix code format
Signed-off-by: Yiheng Wang <vennw@nvidia.com>
---
ci/unit_tests/test_vista3d.py | 2 +-
models/vista3d/scripts/evaluator.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/ci/unit_tests/test_vista3d.py b/ci/unit_tests/test_vista3d.py
index 60f566db..897b207f 100644
--- a/ci/unit_tests/test_vista3d.py
+++ b/ci/unit_tests/test_vista3d.py
@@ -242,7 +242,7 @@
"initialize": ["$monai.utils.set_determinism(seed=123)"],
"error": "Undefined label prompt detected. Provide point prompts for zero-shot.",
}
- ]
+ ],
]
diff --git a/models/vista3d/scripts/evaluator.py b/models/vista3d/scripts/evaluator.py
index ab007c3f..a95bdf30 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 = 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: