diff --git a/mteb/tasks/pair_classification/eng/pub_chem_smilespc.py b/mteb/tasks/pair_classification/eng/pub_chem_smilespc.py index 2b2e1736ab..1f5af6bcb6 100644 --- a/mteb/tasks/pair_classification/eng/pub_chem_smilespc.py +++ b/mteb/tasks/pair_classification/eng/pub_chem_smilespc.py @@ -1,3 +1,5 @@ +from typing import Any + import datasets from mteb.abstasks.pair_classification import AbsTaskPairClassification @@ -74,8 +76,7 @@ class PubChemSMILESPC(AbsTaskPairClassification): """, ) - def load_data(self): - """Load dataset from HuggingFace hub""" + def load_data(self, num_proc: int = 1, **kwargs: Any) -> None: if self.data_loaded: return