forgot pre-commit

dyollb · Nov 7, 2024 · 397ff7c · 397ff7c
1 parent 3645cab
commit 397ff7c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/segmantic/seg/monai_unet.py b/src/segmantic/seg/monai_unet.py
@@ -565,9 +565,7 @@ def predict(
             settings = json.load(json_file)
         net: Net = Net.load_from_checkpoint(f"{model_file}", **settings)
     else:
-        net = Net.load_from_checkpoint(
-            f"{model_file}"
-        )
+        net = Net.load_from_checkpoint(f"{model_file}")
     num_classes = net.num_classes
 
     net.freeze()