diff --git a/mteb/tasks/audio/any_2_any_retrieval/__init__.py b/mteb/tasks/audio/any_2_any_retrieval/__init__.py index c9e776beef..bbaefa47a7 100644 --- a/mteb/tasks/audio/any_2_any_retrieval/__init__.py +++ b/mteb/tasks/audio/any_2_any_retrieval/__init__.py @@ -15,7 +15,7 @@ from .jl_corpus import JLCorpusA2TRetrieval, JLCorpusT2ARetrieval from .libri_tts import LibriTTSA2TRetrieval, LibriTTST2ARetrieval from .macs import MACSA2TRetrieval, MACST2ARetrieval -from .multilingual import * +from .multilingual import JamAltArtist, JamAltLyricsA2T, JamAltLyricsT2A from .music_caps import MusicCapsA2TRetrieval, MusicCapsT2ARetrieval from .sound_descs import SoundDescsA2TRetrieval, SoundDescsT2ARetrieval from .spoken_s_qu_ad import SpokenSQuADT2ARetrieval @@ -44,6 +44,9 @@ "HiFiTTST2ARetrieval", "JLCorpusA2TRetrieval", "JLCorpusT2ARetrieval", + "JamAltArtist", + "JamAltLyricsA2T", + "JamAltLyricsT2A", "LibriTTSA2TRetrieval", "LibriTTST2ARetrieval", "MACSA2TRetrieval", diff --git a/mteb/tasks/audio/audio_zeroshot_classification/eng/__init__.py b/mteb/tasks/audio/audio_zeroshot_classification/eng/__init__.py index 3d10229b13..490bd5ca01 100644 --- a/mteb/tasks/audio/audio_zeroshot_classification/eng/__init__.py +++ b/mteb/tasks/audio/audio_zeroshot_classification/eng/__init__.py @@ -1,11 +1,15 @@ from .esc50 import ESC50ZeroshotClassification from .ravdess import RavdessZeroshotClassification -from .speech_commands import SpeechCommandsZeroshotClassificationV01 +from .speech_commands import ( + SpeechCommandsZeroshotClassificationV01, + SpeechCommandsZeroshotClassificationv02, +) from .urban_sound8k import UrbanSound8kZeroshotClassification __all__ = [ "ESC50ZeroshotClassification", "RavdessZeroshotClassification", "SpeechCommandsZeroshotClassificationV01", + "SpeechCommandsZeroshotClassificationv02", "UrbanSound8kZeroshotClassification", ] diff --git a/tests/test_tasks/test_metadata.py b/tests/test_tasks/test_metadata.py index abe1c8575d..01b9161e03 100644 --- a/tests/test_tasks/test_metadata.py +++ b/tests/test_tasks/test_metadata.py @@ -224,6 +224,7 @@ def test_all_metadata_is_filled_and_valid(task: AbsTask): if task.metadata.name in ( "SpeechCommandsZeroshotv0.01", + "SpeechCommandsZeroshotv0.02", "VoxPopuliGenderClustering", "BirdSet", ):