Oops

stanfordnlp · Nov 19, 2023 · 7f20d53 · 7f20d53
1 parent d490380
commit 7f20d53
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/stanza/utils/datasets/coref/convert_ontonotes.py b/stanza/utils/datasets/coref/convert_ontonotes.py
@@ -70,14 +70,14 @@ def convert_dataset_section(pipe, section):
 
         processed = {
             "document_id": doc_id,
-            "cased_words": sentences,
+            "cased_words": cased_words,
             "sent_id": sent_id,
             "part_id": part_id,
             "speaker": speaker,
             #"pos": pos,
             "deprel": deprel,
             "head": heads,
-            "clusters": clusters
+            "span_clusters": clusters
         }
         processed_section.append(processed)
     return processed_section