fix

ArneBinder · Nov 5, 2024 · 20f3d17 · 20f3d17
1 parent f3cfa0e
commit 20f3d17
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/src/pytorch_ie/pipeline.py b/src/pytorch_ie/pipeline.py
@@ -54,11 +54,7 @@ def __init__(
         **kwargs,
     ):
         self.taskmodule = taskmodule
-        device_str = (
-            torch.device("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