diff --git a/bind/G4Calo.py b/bind/G4Calo.py index eb93c4d..e59c21a 100644 --- a/bind/G4Calo.py +++ b/bind/G4Calo.py @@ -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