assemble steps inside try block to ensure temp file cleanup

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
2026-06-08 14:18:40 +02:00
parent 5f01ab6d35
commit df3faa841c
1 changed files with 4 additions and 2 deletions
@@ -462,15 +462,17 @@ Example
    sw.reset()
    sw.start()
    df = None
+    steps_df = None
    try:
        df = _assemble_results_to_mini_df(rp)
-    finally: #make sure to delete temp files
+        if return_steps:
+            steps_df = _assemble_results_to_mini_df(rp, tree_name="Steps")
+    finally:
        for f in tmpfile:
            if os.path.exists(f):
                os.remove(f)
    print('G4Calo: concatenation finished after {:.2f} seconds'.format(sw.elapsed()))
    if return_steps:
-        steps_df = _assemble_results_to_mini_df(rp, tree_name="Steps")
        return df, steps_df
    return df