From 5a9a00ada6e76fe7212a3410f71a7dc4c20f7381 Mon Sep 17 00:00:00 2001 From: Markus Binsteiner Date: Mon, 12 Aug 2024 13:27:53 +0200 Subject: [PATCH] fix: remove duplicated node_id column if it already exists --- src/kiara_plugin/tropy/models.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/kiara_plugin/tropy/models.py b/src/kiara_plugin/tropy/models.py index a40ca4d..609c72e 100644 --- a/src/kiara_plugin/tropy/models.py +++ b/src/kiara_plugin/tropy/models.py @@ -168,12 +168,18 @@ def create_from_networkx_graph( k: v if v else {"_x_placeholder_x_": "__dummy__"} for k, v in graph.nodes(data=True) } + nodes_data = pd.DataFrame.from_dict(node_dict, orient="index") nodes_data = nodes_data.reset_index() if "_x_placeholder_x_" in nodes_data.columns: nodes_data = nodes_data.drop("_x_placeholder_x_", axis=1) - nodes_data = nodes_data.rename(columns={"index": node_id_column_name}) + + if node_id_column_name in nodes_data.columns: + # remove index column if it exists + nodes_data = nodes_data.drop("index", axis=1) + else: + nodes_data = nodes_data.rename(columns={"index": node_id_column_name}) nodes_table = KiaraTable.create_table(nodes_data)