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
+ 4
4
Compare changes
  • Side-by-side
  • Inline
+ 4
4
@@ -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,7 +147,7 @@ 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._events = None
@@ -268,7 +268,7 @@ class AanetHits:
def __str__(self):
# hits
if all(key.startswith('hits.') for key in self._keys):
return "Number of hits in the selected event: {}".format(len(self))
return "Number of hits: {}".format(len(self))
# mc hits
if all(key.startswith('mc_hits.') for key in self._keys):
return "Number of mc hits in the selected event: {}".format(
Loading