Remove assert on optional arg
This commit is contained in:
@@ -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]
|
||||||
|
|||||||
Reference in New Issue
Block a user