Skip to content

Commit 425a479

Browse files
authored
Fix Datamodule creation for OV in AutoConfigurator (#3920)
Fix datamodule for ov
1 parent f611cc1 commit 425a479

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

src/otx/engine/utils/auto_configurator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -410,6 +410,7 @@ def update_ov_subset_pipeline(self, datamodule: OTXDataModule, subset: str = "te
410410
subset_config.to_tv_image = ov_config[f"{subset}_subset"]["to_tv_image"]
411411
datamodule.image_color_channel = ov_config["image_color_channel"]
412412
datamodule.tile_config.enable_tiler = False
413+
datamodule.unlabeled_subset.data_root = None
413414
msg = (
414415
f"For OpenVINO IR models, Update the following {subset} \n"
415416
f"\t transforms: {subset_config.transforms} \n"

tests/unit/engine/utils/test_auto_configurator.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -212,3 +212,5 @@ def test_update_ov_subset_pipeline(self) -> None:
212212
assert updated_datamodule.test_subset.transforms == [{"class_path": "torchvision.transforms.v2.ToImage"}]
213213

214214
assert updated_datamodule.test_subset.transform_lib_type == TransformLibType.TORCHVISION
215+
assert not updated_datamodule.tile_config.enable_tiler
216+
assert updated_datamodule.unlabeled_subset.data_root is None

0 commit comments

Comments
 (0)