Skip to content

Commit dbbf2b5

Browse files
committed
Wrap lines
1 parent 64812a7 commit dbbf2b5

1 file changed

Lines changed: 11 additions & 6 deletions

File tree

src/graphnet/data/utilities/parquet_to_sqlite.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -100,8 +100,9 @@ def run(self, outdir: str, database_name: str) -> None:
100100
)
101101
self._event_counter += n_events_in_file
102102
self._save_config(outdir, database_name)
103-
print(
104-
f"Database saved at: \n{outdir}/{database_name}/data/{database_name}.db"
103+
self.info(
104+
"Database saved at: \n"
105+
f"{outdir}/{database_name}/data/{database_name}.db"
105106
)
106107

107108
def _count_events(self, open_parquet_file: ak.Array) -> int:
@@ -138,9 +139,10 @@ def _convert_to_dataframe(
138139
if len(df.columns) == 1:
139140
if df.columns == ["values"]:
140141
df.columns = [field_name]
141-
if (
142-
len(df) != n_events_in_file
143-
): # if true, the dataframe contains more than 1 row pr. event (e.g. Pulsemap).
142+
143+
# If true, the dataframe contains more than 1 row pr. event (i.e.,
144+
# pulsemap).
145+
if len(df) != n_events_in_file:
144146
event_nos = []
145147
c = 0
146148
for event_no in range(
@@ -150,7 +152,10 @@ def _convert_to_dataframe(
150152
event_nos.extend(
151153
np.repeat(event_no, len(df[df.columns[0]][c])).tolist()
152154
)
153-
except KeyError: # KeyError indicates that this df has no entry for event_no (e.g. an event with no detector response)
155+
156+
# KeyError indicates that this df has no entry for event_no
157+
# (e.g., an event with no detector response).
158+
except KeyError:
154159
pass
155160
c += 1
156161
else:

0 commit comments

Comments
 (0)