Forward data format to tiling dataset to fix arrow handling

openvinotoolkit · Dec 13, 2024 · da5370e · da5370e
1 parent 2c9b5ec
commit da5370e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/otx/core/data/dataset/tile.py b/src/otx/core/data/dataset/tile.py
@@ -222,6 +222,7 @@ def __init__(self, dataset: OTXDataset, tile_config: TileConfig) -> None:
             dataset.image_color_channel,
             dataset.stack_images,
             dataset.to_tv_image,
+            data_format=dataset.data_format,
         )
         self.tile_config = tile_config
         self._dataset = dataset