From 202b61c4e139ebdfc40af33aac000d703a05e31b Mon Sep 17 00:00:00 2001 From: Ali Shiraee Date: Sun, 25 Jan 2026 19:32:08 +0000 Subject: [PATCH] fix: add num_proc arg to PubChemWikiPairClassification dataset_transform --- .../multilingual/pub_chem_wiki_pair_classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mteb/tasks/pair_classification/multilingual/pub_chem_wiki_pair_classification.py b/mteb/tasks/pair_classification/multilingual/pub_chem_wiki_pair_classification.py index f02cb4db76..23eec9e7d1 100644 --- a/mteb/tasks/pair_classification/multilingual/pub_chem_wiki_pair_classification.py +++ b/mteb/tasks/pair_classification/multilingual/pub_chem_wiki_pair_classification.py @@ -60,7 +60,7 @@ class PubChemWikiPairClassification(AbsTaskPairClassification): """, ) - def dataset_transform(self) -> None: + def dataset_transform(self, num_proc: int = 1) -> None: _dataset = {} for lang in self.hf_subsets: _dataset[lang] = {}