diff --git a/src/firewheel/control/dependency_graph.py b/src/firewheel/control/dependency_graph.py index 031036f4..7e9dc727 100644 --- a/src/firewheel/control/dependency_graph.py +++ b/src/firewheel/control/dependency_graph.py @@ -219,7 +219,7 @@ def get_graph_json(self): {"id": , "type": <"entity" or "constraint">}, ... ], - "links": [ + "edges": [ {"source": , "target": }, ... ], @@ -228,13 +228,13 @@ def get_graph_json(self): "multigraph": False } """ - data = json_graph.node_link_data(self.dg) + data = json_graph.node_link_data(self.dg, edges="edges") nodes = {} for node in data["nodes"]: nodes[node["id"]] = node - for edge in data["links"]: + for edge in data["edges"]: edge["source"] = nodes[edge["source"]]["id"] edge["target"] = nodes[edge["target"]]["id"]