m3hrdadfi commited on
Commit
0e3fc39
1 Parent(s): e5ddbb7

Fix data preparation

Browse files
Files changed (1) hide show
  1. src/run_ed_recipe_nlg.py +4 -0
src/run_ed_recipe_nlg.py CHANGED
@@ -134,6 +134,10 @@ class DataTrainingArguments:
134
  default=None,
135
  metadata={"help": "An optional input evaluation data file to evaluate the perplexity on (a text file)."},
136
  )
 
 
 
 
137
  max_source_length: Optional[int] = field(
138
  default=128,
139
  metadata={
 
134
  default=None,
135
  metadata={"help": "An optional input evaluation data file to evaluate the perplexity on (a text file)."},
136
  )
137
+ test_file: Optional[str] = field(
138
+ default=None,
139
+ metadata={"help": "An optional input evaluation data file to predict the perplexity on (a text file)."},
140
+ )
141
  max_source_length: Optional[int] = field(
142
  default=128,
143
  metadata={