Adjust code style

teamtomo · Oct 30, 2023 · abd7ed2 · abd7ed2
1 parent 26db9de
commit abd7ed2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/membrain_seg/segmentation/segment.py b/src/membrain_seg/segmentation/segment.py
@@ -79,6 +79,7 @@ def segment(
 
     # Initialize the model and load trained weights from checkpoint
     pl_model = SemanticSegmentationUnet.load_from_checkpoint(model_checkpoint, map_location=device, strict=False)
+
     pl_model.to(device)
 
     # Preprocess the new data