From 74723e867828984055c9a76199b2c86a972d735b Mon Sep 17 00:00:00 2001
From: Nathan Voxland <nathan@voxland.net>
Date: Thu, 30 May 2024 11:08:53 -0500
Subject: [PATCH] Store sample_info for segment_mask htypes

---
 deeplake/core/dataset/dataset.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/deeplake/core/dataset/dataset.py b/deeplake/core/dataset/dataset.py
index 8bac10d631..9137bd42f1 100644
--- 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"):