pre-commit

PhoenixDL · Dec 29, 2022 · b9cd7e8 · b9cd7e8
1 parent 8c25169
commit b9cd7e8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/loading/test_loader.py b/tests/loading/test_loader.py
@@ -84,9 +84,9 @@ def test_data_moved_to_gpu(self):
         data = [{"data": 1, "label": 1}, {"data": 2, "label": 2}, {"data": 3, "label": 3}]
         loader = DataLoader(
             data,
-            gpu_transforms=Compose([
-                DeviceChecker(keys=("data",), device="cuda"), DeviceChecker(keys=("label",), device="cpu")]
-        ),
+            gpu_transforms=Compose(
+                [DeviceChecker(keys=("data",), device="cuda"), DeviceChecker(keys=("label",), device="cpu")]
+            ),
             to_gpu_trafo=ToDevice(device="cuda", keys=("data",)),
         )
         for x in loader: