.

harvardnlp · Sep 11, 2019 · 0be222d · 0be222d
1 parent f889527
commit 0be222d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_struct/cky.py b/torch_struct/cky.py
@@ -323,7 +323,7 @@ def to_networkx(cls, spans):
             for k, v in splits[0].items():
                 G.add_edge(indices[k], indices[(b, k[0], v)])
                 G.add_edge(indices[k], indices[(b, v + 1, k[1])])
-        return G
+        return G, indices
 
     ###### Test