Switch back to full corpus aligmnent

matgille · Apr 19, 2024 · 42a8ab3 · 42a8ab3
1 parent 1bbc4f9
commit 42a8ab3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -223,7 +223,7 @@ def run_alignments():
     prefix = args.prefix
     use_punctuation = args.use_punctuation
     print(f"Punctuation for tokenization: {use_punctuation}")
-    MyAligner = Aligner(corpus_size=100, max_align=3, out_dir=out_dir, use_punctuation=use_punctuation, input_dir=input_dir, main_wit=main_wit, prefix=prefix)
+    MyAligner = Aligner(corpus_size=None, max_align=3, out_dir=out_dir, use_punctuation=use_punctuation, input_dir=input_dir, main_wit=main_wit, prefix=prefix)
     MyAligner.parallel_align()
     utils.write_json(f"result_dir/{out_dir}/alignment_dict.json", MyAligner.alignment_dict)
     align_dict = utils.read_json(f"result_dir/{out_dir}/alignment_dict.json")