add brackets

AllenCellModeling · Oct 18, 2023 · 7f3b912 · 7f3b912
1 parent bf19a86
commit 7f3b912
Showing 1 changed file with 8 additions and 6 deletions.
diff --git 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)