Remove assert on optional arg

This commit is contained in:
Sylvain Gugger
2022-01-13 17:34:41 -05:00
parent 1eb40338ac
commit 96881729ce

View File

@@ -241,9 +241,6 @@ class DataTrainingArguments:
def __post_init__(self): def __post_init__(self):
if self.dataset_name is None and self.train_file is None and self.validation_file is None: if self.dataset_name is None and self.train_file is None and self.validation_file is None:
raise ValueError("Need either a dataset name or a training/validation file.") raise ValueError("Need either a dataset name or a training/validation file.")
elif self.lang is None:
raise ValueError("Need to specify the language.")
else: else:
if self.train_file is not None: if self.train_file is not None:
extension = self.train_file.split(".")[-1] extension = self.train_file.split(".")[-1]