Merge pull request #2867 from activeloopai/segment_mask_info

Store sample_info for segment_mask htypes
activeloopai · Jun 3, 2024 · 8836e6c · 8836e6c
2 parents 03e9e26 + 74723e8
commit 8836e6c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deeplake/core/dataset/dataset.py b/deeplake/core/dataset/dataset.py
@@ -908,6 +908,7 @@ def _create_tensor(
             "point_cloud",
             "mesh",
             "nifti",
+            "segment_mask",
         ):
             self._create_sample_info_tensor(name)
         if create_shape_tensor and htype not in ("text", "json", "tag"):