diff --git a/tests/test_alternating_batch_sampler.py b/tests/test_alternating_batch_sampler.py index 48acdb7e0..929590828 100644 --- a/tests/test_alternating_batch_sampler.py +++ b/tests/test_alternating_batch_sampler.py @@ -33,12 +33,14 @@ def __iter__(self): def test_alternating_batch_sampler(shuffle, sampler): root_dir = pyrootutils.find_root() transforms = monai.transforms.Compose( - RemoveNaNKeysd(), - monai.transforms.LoadImaged( - keys=["raw", "seg1", "seg2"], - reader=MonaiBioReader(dimension_order_out="CZYX"), - allow_missing_keys=True, - ), + [ + RemoveNaNKeysd(), + monai.transforms.LoadImaged( + keys=["raw", "seg1", "seg2"], + reader=MonaiBioReader(dimension_order_out="CZYX"), + allow_missing_keys=True, + ), + ] ) transform_dict = {key: transforms for key in ["train", "test", "val", "predict"]} data = make_multiple_dataframe_splits(root_dir / "tests" / "resources", transform_dict)