Avoid deprecation warning when initializing the busmap

openego · Aug 15, 2024 · bfb932d · bfb932d
1 parent 85ac40c
commit bfb932d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/etrago/cluster/spatial.py b/etrago/cluster/spatial.py
@@ -759,7 +759,7 @@ def kmedoids_dijkstra_clustering(
         kmeans.fit(points)
 
         busmap = pd.Series(
-            data=kmeans.predict(buses.loc[buses_i, ["x", "y"]]),
+            data=kmeans.predict(buses.loc[buses_i, ["x", "y"]].values),
             index=buses_i,
             dtype=object,
         )