.

harvardnlp · Sep 11, 2019 · dfcfec0 · dfcfec0
1 parent 6ca3dfc
commit dfcfec0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_struct/cky.py b/torch_struct/cky.py
@@ -321,8 +321,8 @@ def to_networkx(cls, spans):
                 G.add_node(cur, label=n[2].item())
                 cur += 1
         for k, v in splits[0].items():
-            G.add_edge(indices[k], indices[(k[0], k[1], v)])
-            G.add_edge(indices[k], indices[(k[0], v+1, k[2])])
+            G.add_edge(indices[(k[0], k[1], v)], indices[k])
+            G.add_edge(indices[(k[0], v+1, k[2])], indices[k])
         return G, indices
 
     ###### Test