fix_future_warning (#10)

* fix_future_warning --------- Co-authored-by: pascalgutjahr <[email protected]>
icecube · Oct 22, 2024 · f2d0832 · f2d0832
1 parent e2e26e8
commit f2d0832
Showing 1 changed file with 8 additions and 1 deletion.
diff --git 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][0], row[1][1], row[1][2], row[1][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(