Merge pull request #248 from pyplati/nnunet-service-fix-label-save

Save the label mask in the loop
pyplati · Dec 15, 2023 · b4e6eeb · b4e6eeb
2 parents ba9bc08 + 2890943
commit b4e6eeb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/services/nnunet/service.py b/services/nnunet/service.py
@@ -159,10 +159,10 @@ def nnunet_service(data_objects, working_dir, settings):
 
                 sitk.WriteImage(mask, str(mask_file))
 
-            output_data_object = DataObject(
-                type="FILE", path=str(mask_file), parent=data_object
-            )
-            output_objects.append(output_data_object)
+                output_data_object = DataObject(
+                    type="FILE", path=str(mask_file), parent=data_object
+                )
+                output_objects.append(output_data_object)
 
         os.remove(io_path)