Skip to content

Commit 31b2504

Browse files
committed
fix unit test
1 parent cb3d903 commit 31b2504

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

src/otx/tools/converter.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -448,6 +448,9 @@ def instantiate(
448448
model_parser.add_subclass_arguments(OTXModel, "model", required=False, fail_untyped=False, skip={"label_info"})
449449
model = model_parser.instantiate_classes(Namespace(model=model_config)).get("model")
450450

451+
if hasattr(model, "tile_config"):
452+
model.tile_config = datamodule.tile_config
453+
451454
# Instantiate Engine
452455
config_work_dir = config.pop("work_dir", config["engine"].pop("work_dir", None))
453456
config["engine"]["work_dir"] = work_dir if work_dir is not None else config_work_dir

tests/unit/core/data/test_tiling.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -256,7 +256,7 @@ def test_tile_transform(self, task, fxt_data_roots):
256256
dataset = DmDataset.import_from(data_root, format=dataset_format)
257257

258258
rng = np.random.default_rng()
259-
tile_size = rng.integers(low=100, high=500, size=(2,))
259+
tile_size = rng.integers(low=50, high=128, size=(2,))
260260
overlap = rng.random(2)
261261
overlap = overlap.clip(0, 0.9)
262262
threshold_drop_ann = rng.random()

0 commit comments

Comments
 (0)