diff --git a/km3io/rootio.py b/km3io/rootio.py index 77926e03025496dcfd0a07243677b710c02ebf70..f0b9e3f8aa9a9a74239c0a67d89c161d83a23e5e 100644 --- a/km3io/rootio.py +++ b/km3io/rootio.py @@ -28,6 +28,7 @@ class EventReader: step_size=2000, keys=None, aliases=None, + nested_branches=None, event_ctor=None, ): """EventReader base class @@ -65,6 +66,11 @@ class EventReader: self._event_ctor = event_ctor self._index_chain = [] if index_chain is None else index_chain + if aliases is not None: + self.aliases = aliases + if nested_branches is not None: + self.nested_branches = nested_branches + if self._keys is None: self._initialise_keys() @@ -141,6 +147,7 @@ class EventReader: index_chain=self._index_chain + [key], step_size=self._step_size, aliases=self.aliases, + nested_branches=self.nested_branches, keys=self.keys(), event_ctor=self._event_ctor, )