Spaces:
Running
Running
Pedro Cuenca
commited on
Commit
•
a104edb
1
Parent(s):
6c27b0d
accept tsv files as inputs.
Browse files
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
|
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
|
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 |
|