Pedro Cuenca commited on
Commit
a104edb
1 Parent(s): 6c27b0d

accept tsv files as inputs.

Browse files
Files changed (1) hide show
  1. seq2seq/run_seq2seq_flax.py +2 -2
seq2seq/run_seq2seq_flax.py CHANGED
@@ -222,10 +222,10 @@ class DataTrainingArguments:
222
  else:
223
  if self.train_file is not None:
224
  extension = self.train_file.split(".")[-1]
225
- assert extension in ["csv", "json"], "`train_file` should be a csv or a json file."
226
  if self.validation_file is not None:
227
  extension = self.validation_file.split(".")[-1]
228
- assert extension in ["csv", "json"], "`validation_file` should be a csv or a json file."
229
  if self.val_max_target_length is None:
230
  self.val_max_target_length = self.max_target_length
231
 
 
222
  else:
223
  if self.train_file is not None:
224
  extension = self.train_file.split(".")[-1]
225
+ assert extension in ["tsv", "csv", "json"], "`train_file` should be a tsv, csv or json file."
226
  if self.validation_file is not None:
227
  extension = self.validation_file.split(".")[-1]
228
+ assert extension in ["tsv", "csv", "json"], "`validation_file` should be a tsv, csv or json file."
229
  if self.val_max_target_length is None:
230
  self.val_max_target_length = self.max_target_length
231