Skip to content

Commit

Permalink
Add unit-tests
Browse files Browse the repository at this point in the history
  • Loading branch information
harimkang committed Sep 2, 2024
1 parent 7fe1aef commit 6fb8ad5
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions tests/unit/engine/utils/test_auto_configurator.py
Original file line number Diff line number Diff line change
Expand Up @@ -212,3 +212,5 @@ def test_update_ov_subset_pipeline(self) -> None:
assert updated_datamodule.test_subset.transforms == [{"class_path": "torchvision.transforms.v2.ToImage"}]

assert updated_datamodule.test_subset.transform_lib_type == TransformLibType.TORCHVISION
assert not updated_datamodule.tile_config.enable_tiler
assert updated_datamodule.unlabeled_subset.data_root is None

0 comments on commit 6fb8ad5

Please sign in to comment.