black reformatted

Mohinta2892 · Nov 2, 2023 · 0fd3bff · 0fd3bff
1 parent 6f6ff2d
commit 0fd3bff
Showing 1 changed file with 11 additions and 9 deletions.
diff --git a/gunpowder/torch/nodes/predict.py b/gunpowder/torch/nodes/predict.py
@@ -56,14 +56,14 @@ class Predict(GenericPredict):
     """
 
     def __init__(
-            self,
-            model,
-            inputs: Dict[str, ArrayKey],
-            outputs: Dict[Union[str, int], ArrayKey],
-            array_specs: Dict[ArrayKey, ArraySpec] = None,
-            checkpoint: str = None,
-            device="cuda",
-            spawn_subprocess=False,
+        self,
+        model,
+        inputs: Dict[str, ArrayKey],
+        outputs: Dict[Union[str, int], ArrayKey],
+        array_specs: Dict[ArrayKey, ArraySpec] = None,
+        checkpoint: str = None,
+        device="cuda",
+        spawn_subprocess=False,
     ):
         self.array_specs = array_specs if array_specs is not None else {}
 
@@ -87,7 +87,9 @@ def __init__(
 
     def start(self):
         # Issue #188
-        self.use_cuda = torch.cuda.is_available() and self.device_string.__contains__("cuda")
+        self.use_cuda = torch.cuda.is_available() and self.device_string.__contains__(
+            "cuda"
+        )
 
         logger.info(f"Predicting on {'gpu' if self.use_cuda else 'cpu'}")
         self.device = torch.device(self.device_string if self.use_cuda else "cpu")