def _create_edge_map(graph: nx.MultiDiGraph) -> Dict: edge_map = {} for edge in graph.edges(): edge_map[graph[edge[0]][edge[1]][0]["id"]] = edge return edge_map