updated the output of the etl_property_neighbour.py

closes issue #5
2025-01-09 18:22:55 +01:00 · 2025-01-09 18:22:55 +01:00 · cb6935b60c
parent 60a3d7d9b3
commit cb6935b60c
1 changed files with 3 additions and 2 deletions
--- a/etl/src/data/etl_property_neighbours.py
+++ b/etl/src/data/etl_property_neighbours.py
@ -59,6 +59,7 @@ def property_neighbours(id: int):
    extractions = extractions.sort("distances").head(10)
    extractions = extractions.drop('distances')

-    result = {"ids": extractions['id'].to_list(), "lat": extractions['lat'].to_list(), "lon": extractions['lon'].to_list()}
+    #result = {"ids": extractions['id'].to_list(), "lat": extractions['lat'].to_list(), "lon": extractions['lon'].to_list()}
+    result = extractions.to_dicts()

-    return result
+    return result