From da5370e508f339007c406244fcedbe87182d69d4 Mon Sep 17 00:00:00 2001 From: Vladisalv Sovrasov Date: Fri, 13 Dec 2024 22:29:53 +0900 Subject: [PATCH] Forward data format to tiling dataset to fix arrow handling --- src/otx/core/data/dataset/tile.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/otx/core/data/dataset/tile.py b/src/otx/core/data/dataset/tile.py index 4187935aa4..af9810eba0 100644 --- 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