Fix tests

huggingface · Sep 30, 2022 · fd38197 · fd38197 · github-actions · Sep 30, 2022
1 parent b4fcc11
commit fd38197
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/docs/source/package_reference/task_templates.mdx b/docs/source/package_reference/task_templates.mdx
@@ -4,6 +4,8 @@ The tasks supported by [`Dataset.prepare_for_task`] and [`DatasetDict.prepare_fo
 
 [[autodoc]] datasets.tasks.AutomaticSpeechRecognition
 
+[[autodoc]] datasets.tasks.AudioClassification
+
 [[autodoc]] datasets.tasks.ImageClassification
     - align_with_features
 

diff --git a/tests/packaged_modules/test_folder_based_builder.py b/tests/packaged_modules/test_folder_based_builder.py
@@ -18,6 +18,7 @@ class DummyFolderBasedBuilder(FolderBasedBuilder):
     BASE_COLUMN_NAME = "base"
     BUILDER_CONFIG_CLASS = FolderBasedBuilderConfig
     EXTENSIONS = [".txt"]
+    CLASSIFICATION_TASK = None
 
 
 @pytest.fixture