.

harvardnlp · Sep 11, 2019 · b536a40 · b536a40
1 parent 705619b
commit b536a40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_struct/cky.py b/torch_struct/cky.py
@@ -319,7 +319,7 @@ def to_networkx(cls, spans):
         indices = {}
 
         for n in spans.nonzero():
-            indices[(n[0].item(), n[1]].item(), n[2].item())] = cur
+            indices[(n[0].item(), n[1].item(), n[2].item())] = cur
             G.add_node(cur, label=n[3].item())
             cur += 1
         for k, v in splits.items():