Update candidate.py

cosbidev · Oct 19, 2022 · b3f1516 · b3f1516
1 parent ad5107b
commit b3f1516
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pytrack/matching/candidate.py b/pytrack/matching/candidate.py
@@ -63,7 +63,11 @@ def get_candidates(G, points, interp_dist=1, closest=True, radius=10):
     G = G.copy()
 
     if interp_dist:
-        G = distance.interpolate_graph(G, dist=interp_dist)
+        if G.graph["geometry"]:
+            G = distance.interpolate_graph(G, dist=interp_dist)
+        else:
+            _ = utils.graph_to_gdfs(G, nodes=False)
+            G = distance.interpolate_graph(G, dist=interp_dist)
 
     geoms = utils.graph_to_gdfs(G, nodes=False).set_index(["u", "v"])[["osmid", "geometry"]]