Store sample_info for segment_mask htypes

activeloopai · May 30, 2024 · 74723e8 · 74723e8
1 parent 03e9e26
commit 74723e8
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"):