diff --git a/dnn_reco/data_handler.py b/dnn_reco/data_handler.py
index 8959044..f3cefe7 100644
--- a/dnn_reco/data_handler.py
+++ b/dnn_reco/data_handler.py
@@ -291,7 +291,14 @@ def read_icecube_data(
         size = len(_time_range["Event"])
         eventIDDict = {}
         for row in _time_range.iterrows():
-            eventIDDict[(row[1].iloc[0], row[1].iloc[1], row[1].iloc[2], row[1].iloc[3])] = row[0]
+            eventIDDict[
+                (
+                    row[1].iloc[0],
+                    row[1].iloc[1],
+                    row[1].iloc[2],
+                    row[1].iloc[3],
+                )
+            ] = row[0]
 
         # Create arrays for input data
         x_ic78 = np.ones(