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