From cb6935b60cdc1e83ecc6e31b7fbd70a7679a548f Mon Sep 17 00:00:00 2001
From: mmaurostoffel <166130318+mmaurostoffel@users.noreply.github.com>
Date: Thu, 9 Jan 2025 18:22:55 +0100
Subject: [PATCH] updated the output of the etl_property_neighbour.py

closes issue #5
---
 etl/src/data/etl_property_neighbours.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/etl/src/data/etl_property_neighbours.py b/etl/src/data/etl_property_neighbours.py
index ccd3ec7..7a4c707 100644
--- 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
\ No newline at end of file