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
+ 6
6
Compare changes
  • Side-by-side
  • Inline
+ 6
6
@@ -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,14 +161,14 @@ 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):
if self._events is None:
self._events = AanetEvents(
self.events_keys,
[self._lazy_data[key] for key in self.events_keys])
[self._data[key] for key in self.events_keys])
return self._events
@property
@@ -176,7 +176,7 @@ class AanetReader(AanetKeys):
if self._hits is None:
self._hits = AanetHits(
self.hits_keys,
[self._lazy_data[key] for key in self.hits_keys])
[self._data[key] for key in self.hits_keys])
return self._hits
@property
Loading