blacked dynunet_wrapper

mlcommons · Oct 4, 2024 · 751325d · 751325d
1 parent 6468f12
commit 751325d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/GANDLF/models/dynunet_wrapper.py b/GANDLF/models/dynunet_wrapper.py
@@ -64,7 +64,8 @@ def __init__(self, parameters: dict):
 
         patch_size = parameters.get("patch_size", None)
         spacing = parameters.get(
-            "spacing_for_internal_computations", [1.0 for i in range(parameters["model"]["dimension"])]
+            "spacing_for_internal_computations",
+            [1.0 for i in range(parameters["model"]["dimension"])],
         )
         parameters["model"]["kernel_size"] = parameters["model"].get(
             "kernel_size", None