Minor fix

mariosasko · mariosasko · commit bf783f2783cc · 2022-09-30T18:31:42.000+02:00
diff --git a/src/datasets/packaged_modules/folder_based_builder/folder_based_builder.py b/src/datasets/packaged_modules/folder_based_builder/folder_based_builder.py
@@ -217,7 +217,7 @@ def analyze(files_or_archives, downloaded_files_or_dirs, split):
                         "label": datasets.ClassLabel(names=sorted(labels)),
                     }
                 )
-                self.info.task_templates = [self.CLASSIFICATION_TASK]
+                self.info.task_templates = [self.CLASSIFICATION_TASK.align_with_features(self.info.features)]
             else:
                 self.info.features = datasets.Features({self.BASE_COLUMN_NAME: self.BASE_FEATURE})
 

Original file line number	Diff line number	Diff line change
`@@ -217,7 +217,7 @@ def analyze(files_or_archives, downloaded_files_or_dirs, split):`
`217`	`217`	`"label": datasets.ClassLabel(names=sorted(labels)),`
`218`	`218`	`}`
`219`	`219`	`)`
`220`		`- self.info.task_templates = [self.CLASSIFICATION_TASK]`
	`220`	`+ self.info.task_templates = [self.CLASSIFICATION_TASK.align_with_features(self.info.features)]`
`221`	`221`	`else:`
`222`	`222`	`self.info.features = datasets.Features({self.BASE_COLUMN_NAME: self.BASE_FEATURE})`
`223`	`223`