added test file (#12630)

2021-07-12 12:15:14 +05:30
parent 9ee66adadb
commit de23ecea36
1 changed files with 4 additions and 0 deletions
--- a/examples/flax/summarization/run_summarization_flax.py
+++ b/examples/flax/summarization/run_summarization_flax.py
@@ -135,6 +135,10 @@ class DataTrainingArguments:
        default=None,
        metadata={"help": "An optional input evaluation data file to evaluate the perplexity on (a text file)."},
    )
+    test_file: Optional[str] = field(
+        default=None,
+        metadata={"help": "An optional input predict data file to do prediction on (a text file)."},
+    )
    max_source_length: Optional[int] = field(
        default=1024,
        metadata={