Fix bug

aistairc · Aug 12, 2020 · 005976a · 005976a
1 parent c849a8d
commit 005976a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/scripts/preprocess.py b/scripts/preprocess.py
@@ -610,7 +610,7 @@ def convert(corpus_name, corpus_dir, output_dir):
 
         a1_file = basename + ".a1"
         if os.path.exists(a1_file):
-            original_doc, a1_entities, a1_relations, a1_events, a1_modalities, a1_attributes, a1_equivalences = parse_standoff_file(
+            _, a1_entities, a1_relations, a1_events, a1_modalities, a1_attributes, a1_equivalences = parse_standoff_file(
                 a1_file, fn, encoding="UTF-8"
             )
             entities.update(a1_entities)
@@ -640,6 +640,8 @@ def convert(corpus_name, corpus_dir, output_dir):
             output_dir, corpus_name, os.path.relpath(basename, corpus_dir)
         )
 
+        original_doc = read_text(fn)
+
         write_text(original_doc, doc_fn + ".txt.ori")
 
         cursor = 0