diff --git a/docs/source/package_reference/task_templates.mdx b/docs/source/package_reference/task_templates.mdx index ec8bcae0217..52d275b8531 100644 --- 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 index 51b5d46d8f9..a7017ea054c 100644 --- 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