fix: remove duplicated node_id column if it already exists

DHARPA-Project · Aug 12, 2024 · 5a9a00a · 5a9a00a
1 parent 6c88140
commit 5a9a00a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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)