Pass arguments of get_h3_indices to h3 function

MTES-MCT · Dec 20, 2024 · 640ff2a · 640ff2a
1 parent 005878d
commit 640ff2a
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/datascience/src/pipeline/helpers/spatial.py b/datascience/src/pipeline/helpers/spatial.py
@@ -214,9 +214,7 @@ def get_h3_indices(
         res = pd.Series([], dtype=object)
     else:
         res = df.apply(
-            lambda row: h3.latlng_to_cell(
-                row["latitude"], row["longitude"], resolution
-            ),
+            lambda row: h3.latlng_to_cell(row[lat], row[lon], resolution),
             axis=1,
         )