From 6568b02f3ad46f7735a6daba6b5f2af9eeca3537 Mon Sep 17 00:00:00 2001 From: Philipp Horstenkamp Date: Sat, 6 Jan 2024 18:29:50 +0100 Subject: [PATCH] Removed the edge color extra definition (#532) --- .../utils/networkx/network_2d.py | 9 --------- .../utils/networkx/network_3d.py | 8 +------- 2 files changed, 1 insertion(+), 16 deletions(-) diff --git a/src/aki_prj23_transparenzregister/utils/networkx/network_2d.py b/src/aki_prj23_transparenzregister/utils/networkx/network_2d.py index 062784f..8bd993d 100644 --- a/src/aki_prj23_transparenzregister/utils/networkx/network_2d.py +++ b/src/aki_prj23_transparenzregister/utils/networkx/network_2d.py @@ -109,15 +109,6 @@ def create_2d_graph( # noqa PLR0913 # ids = list(nx.get_node_attributes(graph, "id").values()) # # Get the Node Text - # node_names = [] - # for key, value in nodes.items(): - # if "name" in value: - # node_names.append(value["name"]) - # else: - # node_names.append(value["firstname"] + " " + value["lastname"]) - # # Add Color and Names to the Scatter Plot. - # print(colors) - # print(node_names) node_trace.marker.color = colors node_trace.marker.line = {"color": "#2e2c2f", "width": 0.5} node_trace.text = node_names diff --git a/src/aki_prj23_transparenzregister/utils/networkx/network_3d.py b/src/aki_prj23_transparenzregister/utils/networkx/network_3d.py index 1e1daec..e821a77 100644 --- a/src/aki_prj23_transparenzregister/utils/networkx/network_3d.py +++ b/src/aki_prj23_transparenzregister/utils/networkx/network_3d.py @@ -177,13 +177,7 @@ def create_3d_graph( # noqa : PLR0913 node_trace.marker.size = list(np.cbrt(metrics[metric]) * 200) # Set the Color and width of Edges for better highlighting. - edge_colors = [] - for row in edges: - if row["type"] == "HAFTENDER_GESELLSCHAFTER": - edge_colors.append("rgb(255,0,0)") - else: - edge_colors.append("rgb(255,105,180)") - edge_trace.line = {"color": edge_colors, "width": edge_thickness} + edge_trace.line = {"color": "rgb(255,105,180)", "width": edge_thickness} # Add Relation_Type as a Description for the edges. if edge_annotation: