Skip to content

Commit 2a6bd71

Browse files
author
miranov25
committed
adding metadata to all trees
1 parent 64b27cb commit 2a6bd71

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

UTILS/dfextensions/AliasDataFrame.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -300,6 +300,8 @@ def export_tree(self, filename_or_file, treename="tree", dropAliasColumns=True):
300300
self._write_metadata_to_root(filename_or_file, treename)
301301
else:
302302
self._write_to_uproot(filename_or_file, treename, dropAliasColumns)
303+
for subframe_name, sub_adf in self._subframes.items():
304+
sub_adf._write_metadata_to_root(filename_or_file, f"{treename}__subframe__{subframe_name}")
303305

304306
def _write_to_uproot(self, uproot_file, treename, dropAliasColumns):
305307
export_cols = [col for col in self.df.columns if not dropAliasColumns or col not in self.aliases]

0 commit comments

Comments
 (0)