From 8975ed652776a1abc6bab2b93fe45f5e19ce87e8 Mon Sep 17 00:00:00 2001
From: Arne Binder <arne.binder@dfki.de>
Date: Tue, 5 Nov 2024 11:23:39 +0100
Subject: [PATCH] make pre-commit happy

---
 src/pytorch_ie/pipeline.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/pytorch_ie/pipeline.py b/src/pytorch_ie/pipeline.py
index 05383acb..e80d2658 100644
--- 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