Joshua Lochner commited on
Commit
bb853de
1 Parent(s): eaa79a8

Fix duplicate argument in function definition

Browse files
Files changed (1) hide show
  1. src/train_classifier.py +4 -6
src/train_classifier.py CHANGED
@@ -48,12 +48,10 @@ class ClassifierTrainingArguments(CustomTrainingArguments, TrainingArguments):
48
 
49
  @dataclass
50
  class ClassifierDatasetArguments(DatasetArguments):
51
- train_file: Optional[str] = DatasetArguments.__dataclass_fields__[
52
- 'c_train_file']
53
- validation_file: Optional[str] = DatasetArguments.__dataclass_fields__[
54
- 'c_validation_file']
55
- test_file: Optional[str] = DatasetArguments.__dataclass_fields__[
56
- 'c_test_file']
57
 
58
 
59
  def main():
 
48
 
49
  @dataclass
50
  class ClassifierDatasetArguments(DatasetArguments):
51
+ def __post_init__(self):
52
+ self.train_file = self.c_train_file
53
+ self.validation_file = self.c_validation_file
54
+ self.test_file = self.c_test_file
 
 
55
 
56
 
57
  def main():