Skip to content
Snippets Groups Projects

Api update

Merged Zineb Aly requested to merge api-update into master
7 unresolved threads
Compare and Show latest version
1 file
+ 5
5
Compare changes
  • Side-by-side
  • Inline
+ 5
5
@@ -102,7 +102,7 @@ class Reader(AanetKeys):
path-like object that points to the file of ineterst.
"""
super().__init__(file_path)
self._lazy_data = uproot.open(self._file_path)['E'].lazyarrays()
self._data = uproot.open(self._file_path)['E'].lazyarrays()
def __getitem__(self, key):
"""reads data stored in the branch of interest in an event tree.
@@ -132,7 +132,7 @@ class Reader(AanetKeys):
if key not in self.valid_keys and not isinstance(key, int):
raise KeyError(
"'{}' is not a valid key or is a fake branch.".format(key))
return self._lazy_data[key]
return self._data[key]
class AanetReader(AanetKeys):
@@ -147,9 +147,9 @@ class AanetReader(AanetKeys):
"""
super().__init__(file_path)
if data is not None:
self._lazy_data = data
self._data = data
else:
self._lazy_data = uproot.open(self._file_path)['E'].lazyarrays()
self._data = uproot.open(self._file_path)['E'].lazyarrays()
self._events = None
self._hits = None
self._tracks = None
@@ -161,7 +161,7 @@ class AanetReader(AanetKeys):
def __getitem__(self, item):
return AanetReader(file_path=self._file_path,
data=self._lazy_data[item])
data=self._data[item])
@property
def events(self):
Loading