darabos commited on
Commit
6373590
·
1 Parent(s): 2b83b53

Include edge attributes in to_nx(), so e.g. Cypher can refer to them.

Browse files
lynxkite-graph-analytics/src/lynxkite_graph_analytics/lynxkite_ops.py CHANGED
@@ -84,8 +84,16 @@ class Bundle:
84
  if df.index.name != "id":
85
  df = df.set_index("id")
86
  graph.add_nodes_from(df.to_dict("index").items())
 
87
  graph.add_edges_from(
88
- self.dfs["edges"][["source", "target"]].itertuples(index=False, name=None)
 
 
 
 
 
 
 
89
  )
90
  return graph
91
 
 
84
  if df.index.name != "id":
85
  df = df.set_index("id")
86
  graph.add_nodes_from(df.to_dict("index").items())
87
+ edges = self.dfs["edges"]
88
  graph.add_edges_from(
89
+ [
90
+ (
91
+ e["source"],
92
+ e["target"],
93
+ {k: e[k] for k in edges.columns if k not in ["source", "target"]},
94
+ )
95
+ for e in edges.to_records()
96
+ ]
97
  )
98
  return graph
99