feat: ✨ Add safe failing for segment blockwise

janelia-cellmap · Apr 17, 2024 · e41a9a3 · e41a9a3
1 parent 049e448
commit e41a9a3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dacapo/blockwise/scheduler.py b/dacapo/blockwise/scheduler.py
@@ -197,5 +197,10 @@ def segment_blockwise(
 
     success = success and daisy.run_blockwise([task])
 
-    shutil.rmtree(tmpdir, ignore_errors=True)
+    if success:
+        shutil.rmtree(tmpdir, ignore_errors=True)
+    else:
+        raise RuntimeError(
+            f"Blockwise segmentation failed. Can rerun with merge files stored at:\n\t{tmpdir}"
+        )
     return success