Fix arrow support for semantic segmentation task (#4172)

Fix arrow support for sseg
openvinotoolkit · Jan 7, 2025 · 4d3df8c · 4d3df8c
1 parent 8296a43
commit 4d3df8c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/otx/core/data/dataset/segmentation.py b/src/otx/core/data/dataset/segmentation.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2023 Intel Corporation
+# Copyright (C) 2023-2025 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 #
 """Module for OTXSegmentationDataset."""
@@ -179,6 +179,7 @@ def __init__(
             image_color_channel,
             stack_images,
             to_tv_image,
+            data_format=data_format,
         )
 
         if self.has_polygons: