fix file closing not to overwrite files when running in chunked mode (#…

…100)
LSSTDESC · Apr 3, 2024 · ebdc0c3 · ebdc0c3
1 parent 9dd9522
commit ebdc0c3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/rail/core/stage.py b/src/rail/core/stage.py
@@ -396,7 +396,8 @@ def _finalize_tag(self, tag):
         """
         handle = self.get_handle(tag, allow_missing=True)
         if self.config.output_mode == 'default':
-            handle.write()
+            if not os.path.exists(handle.path) or not handle.partial:
+                handle.write()            
         final_name = PipelineStage._finalize_tag(self, tag)
         handle.path = final_name
         return final_name