Skip to content

Commit 411a4ce

Browse files
isaac-chungclaude
andcommitted
Fix JamAlt task class naming to match metadata names
Renamed classes to match their metadata names so they can be found in the task registry: - JamAltArtist → JamAltArtistA2ARetrieval - JamAltLyricsT2A → JamAltLyricT2ARetrieval - JamAltLyricsA2T → JamAltLyricA2TRetrieval Also added explicit imports and exports for proper registration. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <[email protected]>
1 parent fb7061a commit 411a4ce

File tree

3 files changed

+19
-8
lines changed

3 files changed

+19
-8
lines changed

mteb/tasks/audio/any_2_any_retrieval/__init__.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,11 @@
1515
from .jl_corpus import JLCorpusA2TRetrieval, JLCorpusT2ARetrieval
1616
from .libri_tts import LibriTTSA2TRetrieval, LibriTTST2ARetrieval
1717
from .macs import MACSA2TRetrieval, MACST2ARetrieval
18-
from .multilingual import *
18+
from .multilingual import (
19+
JamAltArtistA2ARetrieval,
20+
JamAltLyricA2TRetrieval,
21+
JamAltLyricT2ARetrieval,
22+
)
1923
from .music_caps import MusicCapsA2TRetrieval, MusicCapsT2ARetrieval
2024
from .sound_descs import SoundDescsA2TRetrieval, SoundDescsT2ARetrieval
2125
from .spoken_s_qu_ad import SpokenSQuADT2ARetrieval
@@ -44,6 +48,9 @@
4448
"HiFiTTST2ARetrieval",
4549
"JLCorpusA2TRetrieval",
4650
"JLCorpusT2ARetrieval",
51+
"JamAltArtistA2ARetrieval",
52+
"JamAltLyricA2TRetrieval",
53+
"JamAltLyricT2ARetrieval",
4754
"LibriTTSA2TRetrieval",
4855
"LibriTTST2ARetrieval",
4956
"MACSA2TRetrieval",
Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,11 @@
1-
from .jam_alt import JamAltArtist, JamAltLyricsA2T, JamAltLyricsT2A
1+
from .jam_alt import (
2+
JamAltArtistA2ARetrieval,
3+
JamAltLyricA2TRetrieval,
4+
JamAltLyricT2ARetrieval,
5+
)
26

37
__all__ = [
4-
"JamAltArtist",
5-
"JamAltLyricsA2T",
6-
"JamAltLyricsT2A",
8+
"JamAltArtistA2ARetrieval",
9+
"JamAltLyricA2TRetrieval",
10+
"JamAltLyricT2ARetrieval",
711
]

mteb/tasks/audio/any_2_any_retrieval/multilingual/jam_alt.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def _load_jam_alt_data(
117117
return corpus, queries, relevant_docs
118118

119119

120-
class JamAltArtist(AbsTaskRetrieval):
120+
class JamAltArtistA2ARetrieval(AbsTaskRetrieval):
121121
metadata = TaskMetadata(
122122
name="JamAltArtistA2ARetrieval",
123123
description="Given audio clip of a song (query), retrieve all songs from the same artist in the Jam-Alt-Lines dataset",
@@ -178,7 +178,7 @@ def load_data(self, **kwargs):
178178
self.data_loaded = True
179179

180180

181-
class JamAltLyricsT2A(AbsTaskRetrieval):
181+
class JamAltLyricT2ARetrieval(AbsTaskRetrieval):
182182
metadata = TaskMetadata(
183183
name="JamAltLyricT2ARetrieval",
184184
description="From textual lyrics (query), retrieve corresponding audio clips of songs from the Jam-Alt-Lines dataset",
@@ -236,7 +236,7 @@ def load_data(self, **kwargs):
236236
self.data_loaded = True
237237

238238

239-
class JamAltLyricsA2T(AbsTaskRetrieval):
239+
class JamAltLyricA2TRetrieval(AbsTaskRetrieval):
240240
metadata = TaskMetadata(
241241
name="JamAltLyricA2TRetrieval",
242242
description="From audio clips of songs (query), retrieve corresponding textual lyric from the Jam-Alt-Lines dataset",

0 commit comments

Comments
 (0)