Skip to content
Snippets Groups Projects

Best reco hits events

Merged Zineb Aly requested to merge best-reco-hits-events into master
1 unresolved thread
+ 6
4
@@ -568,8 +568,10 @@ class OfflineReader:
str(stages)))
else:
fit_data = np.array([
i[k] for i, j, k in zip(fit_info, rec_stages[:, 0],
rec_stages[:, 1]) if k is not None
i[k]
for i, j, k in zip(fit_info, rec_stages[:, 0], rec_stages[:,
1])
if k is not None
])
rec_array = np.core.records.fromarrays(fit_data.transpose(),
names=keys)
@@ -690,8 +692,8 @@ class OfflineReader:
for key in keys:
lazy_d[key] = np.array([
i[k] for i, k in zip(
getattr(self.tracks, key)[mask], rec_stages[:, 1]
[mask])
getattr(self.tracks, key)[mask], rec_stages[:,
1][mask])
])
return lazy_d
Loading