From 887d7d04a9dcecea93c1d871f9ea3b52601c46aa Mon Sep 17 00:00:00 2001
From: zineb aly <aly.zineb.az@gmail.com>
Date: Tue, 18 Feb 2020 16:36:47 +0100
Subject: [PATCH] correct code style
---
km3io/offline.py | 12 +++++-------
1 file changed, 5 insertions(+), 7 deletions(-)
diff --git a/km3io/offline.py b/km3io/offline.py
index 0333684..8bc3db1 100644
--- a/km3io/offline.py
+++ b/km3io/offline.py
@@ -143,9 +143,9 @@ class OfflineKeys:
except those found in fake branches.
"""
if self._mc_tracks_keys is None:
- fake_branches = set([
- 'mc_trks.usr_data', 'mc_trks.usr', 'mc_trks.usr_names'
- ]) # same solution as above can be used
+ fake_branches = set(
+ ['mc_trks.usr_data', 'mc_trks.usr',
+ 'mc_trks.usr_names']) # same solution as above can be used
tree = uproot.open(self._file_path)['E']['Evt']['mc_trks']
self._mc_tracks_keys = [
key.decode('utf8') for key in tree.keys()
@@ -568,10 +568,8 @@ 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)
--
GitLab