Adjust to precommit style

teamtomo · Oct 30, 2023 · 10da629 · 10da629
1 parent abd7ed2
commit 10da629
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/membrain_seg/segmentation/segment.py b/src/membrain_seg/segmentation/segment.py
@@ -78,8 +78,9 @@ def segment(
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     # Initialize the model and load trained weights from checkpoint
-    pl_model = SemanticSegmentationUnet.load_from_checkpoint(model_checkpoint, map_location=device, strict=False)
-
+    pl_model = SemanticSegmentationUnet.load_from_checkpoint(
+        model_checkpoint, map_location=device, strict=False
+    )
     pl_model.to(device)
 
     # Preprocess the new data