Spaces:
Running
Running
Joshua Lochner
commited on
Commit
•
bb853de
1
Parent(s):
eaa79a8
Fix duplicate argument in function definition
Browse files- 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 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
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():
|