From 3ac93175faf4cdf7b9d37f6aed1ba5df3fbfb7ec Mon Sep 17 00:00:00 2001
From: zineb aly <aly.zineb.az@gmail.com>
Date: Wed, 19 Feb 2020 21:18:22 +0100
Subject: [PATCH] fix code style
---
km3io/offline.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/km3io/offline.py b/km3io/offline.py
index c71b49a..ba4d0e5 100644
--- a/km3io/offline.py
+++ b/km3io/offline.py
@@ -569,8 +569,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)
@@ -691,8 +693,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