From de23ecea36e19ab5184f136a55dcda54d54f74d4 Mon Sep 17 00:00:00 2001 From: Bhadresh Savani Date: Mon, 12 Jul 2021 12:15:14 +0530 Subject: [PATCH] added test file (#12630) --- examples/flax/summarization/run_summarization_flax.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/examples/flax/summarization/run_summarization_flax.py b/examples/flax/summarization/run_summarization_flax.py index 636fa3bb85..60292c4d84 100644 --- 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={