compatible with newer osmnx

BayAreaMetro · Sep 27, 2024 · f21b772 · f21b772
1 parent 955a37c
commit f21b772
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/network_wrangler/roadwaynetwork.py b/network_wrangler/roadwaynetwork.py
@@ -830,6 +830,9 @@ def ox_graph(
 
         WranglerLogger.debug("starting ox.gdfs_to_graph()")
         try:
+            if (int(ox.__version__.split('.')[0]) >= 1):
+                # index required in newer osmnx
+                graph_links = graph_links.set_index(["u", "v", "key"])
             G = ox.graph_from_gdfs(graph_nodes, graph_links)
         except AttributeError:
             WranglerLogger.debug(