refactor: make shorter the uuid length (#81)

Signed-off-by: ktro2828 <[email protected]>
tier4 · Dec 28, 2024 · 91df6a1 · 91df6a1
1 parent d7e8ad7
commit 91df6a1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/t4_devkit/viewer/rendering_data/box.py b/t4_devkit/viewer/rendering_data/box.py
@@ -197,7 +197,7 @@ def _append_with_box(self, box: Box2D) -> None:
         self.class_ids.append(self.label2id[box.semantic_label.name])
 
         if box.uuid is not None:
-            self.uuids.append(box.uuid)
+            self.uuids.append(box.uuid[:6])
 
     def _append_with_elements(self, roi: RoiType, class_id: int, uuid: str | None = None) -> None:
         self.rois.append(roi)

diff --git a/t4_devkit/viewer/rendering_data/segmentation.py b/t4_devkit/viewer/rendering_data/segmentation.py
@@ -53,7 +53,9 @@ def append(self, mask: NDArrayU8, class_id: int, uuid: str | None = None) -> Non
                 )
         self.masks.append(mask)
         self.class_ids.append(class_id)
-        self.uuids.append(uuid)
+
+        if uuid is not None:
+            self.uuids.append(uuid[:6])
 
     def as_segmentation_image(self) -> rr.SegmentationImage:
         """Return mask data as a `rr.SegmentationImage`.