From 5eed17e9fbfa6e4ff0bab57d4fa56ddd10414530 Mon Sep 17 00:00:00 2001
From: Tamas Gal <tgal@km3net.de>
Date: Wed, 19 Feb 2020 15:02:21 +0100
Subject: [PATCH] Make yapf
---
km3io/offline.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/km3io/offline.py b/km3io/offline.py
index 5d55166..8065bcc 100644
--- a/km3io/offline.py
+++ b/km3io/offline.py
@@ -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
--
GitLab