make pre-commit happy

ArneBinder · Nov 5, 2024 · 8975ed6 · 8975ed6
1 parent 20f3d17
commit 8975ed6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/pytorch_ie/pipeline.py b/src/pytorch_ie/pipeline.py
@@ -54,7 +54,9 @@ def __init__(
         **kwargs,
     ):
         self.taskmodule = taskmodule
-        device_str = ("cpu" if device < 0 else f"cuda:{device}") if isinstance(device, int) else device
+        device_str = (
+            ("cpu" if device < 0 else f"cuda:{device}") if isinstance(device, int) else device
+        )
         self.device = torch.device(device_str)
         self.binary_output = binary_output