mteb 2.7.17__py3-none-any.whl → 2.7.19__py3-none-any.whl

mteb/tasks/retrieval/code/code_rag.py

@@ -51,7 +51,7 @@ class CodeRAGProgrammingSolutionsRetrieval(AbsTaskRetrieval):
         **common_args,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         """Load dataset from HuggingFace hub"""
         if self.data_loaded:
             return
@@ -59,7 +59,7 @@ class CodeRAGProgrammingSolutionsRetrieval(AbsTaskRetrieval):
         self.dataset_transform()
         self.data_loaded = True
 
-    def dataset_transform(self, num_proc: int = 1, **kwargs) -> None:
+    def dataset_transform(self, num_proc: int | None = None, **kwargs) -> None:
         """And transform to a retrieval dataset, which have the following attributes
 
         self.corpus = Dict[doc_id, Dict[str, str]] #id => dict with document data like title and text
@@ -108,7 +108,7 @@ class CodeRAGOnlineTutorialsRetrieval(AbsTaskRetrieval):
         **common_args,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         """Load dataset from HuggingFace hub"""
         if self.data_loaded:
             return
@@ -116,7 +116,7 @@ class CodeRAGOnlineTutorialsRetrieval(AbsTaskRetrieval):
         self.dataset_transform()
         self.data_loaded = True
 
-    def dataset_transform(self, num_proc: int = 1, **kwargs) -> None:
+    def dataset_transform(self, num_proc: int | None = None, **kwargs) -> None:
         """And transform to a retrieval dataset, which have the following attributes
 
         self.corpus = Dict[doc_id, Dict[str, str]] #id => dict with document data like title and text
@@ -168,7 +168,7 @@ class CodeRAGLibraryDocumentationSolutionsRetrieval(AbsTaskRetrieval):
         **common_args,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         """Load dataset from HuggingFace hub"""
         if self.data_loaded:
             return
@@ -176,7 +176,7 @@ class CodeRAGLibraryDocumentationSolutionsRetrieval(AbsTaskRetrieval):
         self.dataset_transform()
         self.data_loaded = True
 
-    def dataset_transform(self, num_proc: int = 1, **kwargs) -> None:
+    def dataset_transform(self, num_proc: int | None = None, **kwargs) -> None:
         """And transform to a retrieval dataset, which have the following attributes
 
         self.corpus = Dict[doc_id, Dict[str, str]] #id => dict with document data like title and text
@@ -225,7 +225,7 @@ class CodeRAGStackoverflowPostsRetrieval(AbsTaskRetrieval):
         **common_args,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         """Load dataset from HuggingFace hub"""
         if self.data_loaded:
             return
@@ -233,7 +233,7 @@ class CodeRAGStackoverflowPostsRetrieval(AbsTaskRetrieval):
         self.dataset_transform()
         self.data_loaded = True
 
-    def dataset_transform(self, num_proc: int = 1, **kwargs) -> None:
+    def dataset_transform(self, num_proc: int | None = None, **kwargs) -> None:
         """And transform to a retrieval dataset, which have the following attributes
 
         self.corpus = Dict[doc_id, Dict[str, str]] #id => dict with document data like title and text

mteb/tasks/retrieval/code/code_search_net_cc_retrieval.py

@@ -99,7 +99,7 @@ class CodeSearchNetCCRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/code/coir_code_search_net_retrieval.py

@@ -97,7 +97,7 @@ class COIRCodeSearchNetRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/code/ds1000_retrieval.py

@@ -34,7 +34,7 @@ class DS1000Retrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/code/fresh_stack_retrieval.py

@@ -37,7 +37,7 @@ class FreshStackRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/code/human_eval_retrieval.py

@@ -34,7 +34,7 @@ class HumanEvalRetrieval(AbsTaskRetrieval):
 }""",
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/code/mbpp_retrieval.py

@@ -34,7 +34,7 @@ class MBPPRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/code/wiki_sql_retrieval.py

@@ -36,7 +36,7 @@ class WikiSQLRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/dan/dan_fever_retrieval.py

@@ -47,7 +47,7 @@ Derczynski, Leon},
         task_subtypes=["Claim verification"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         """Load dataset from HuggingFace hub"""
         if self.data_loaded:
             return
@@ -55,7 +55,7 @@ Derczynski, Leon},
         self.dataset_transform()
         self.data_loaded = True
 
-    def dataset_transform(self, num_proc: int = 1, **kwargs) -> None:
+    def dataset_transform(self, num_proc: int | None = None, **kwargs) -> None:
         """And transform to a retrieval dataset, which have the following attributes
 
         self.corpus = dict[doc_id, dict[str, str]] #id => dict with document data like title and text

mteb/tasks/retrieval/dan/tv2_nordretrieval.py

@@ -60,7 +60,7 @@ Piperidis, Stelios},
         task_subtypes=["Article retrieval"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         """Load dataset from HuggingFace hub"""
         if self.data_loaded:
             return
@@ -68,7 +68,7 @@ Piperidis, Stelios},
         self.dataset_transform()
         self.data_loaded = True
 
-    def dataset_transform(self, num_proc: int = 1, **kwargs) -> None:
+    def dataset_transform(self, num_proc: int | None = None, **kwargs) -> None:
         """And transform to a retrieval dataset, which have the following attributes
 
         self.corpus = dict[doc_id, dict[str, str]] #id => dict with document data like title and text

mteb/tasks/retrieval/dan/twitter_hjerne_retrieval.py

@@ -36,7 +36,7 @@ class TwitterHjerneRetrieval(AbsTaskRetrieval):
         task_subtypes=["Question answering"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         """Load dataset from HuggingFace hub"""
         if self.data_loaded:
             return
@@ -44,7 +44,7 @@ class TwitterHjerneRetrieval(AbsTaskRetrieval):
         self.dataset_transform()
         self.data_loaded = True
 
-    def dataset_transform(self, num_proc: int = 1, **kwargs) -> None:
+    def dataset_transform(self, num_proc: int | None = None, **kwargs) -> None:
         """And transform to a retrieval dataset, which have the following attributes
 
         self.corpus = dict[doc_id, dict[str, str]] #id => dict with document data like title and text

mteb/tasks/retrieval/deu/german_gov_service_retrieval.py

@@ -50,7 +50,7 @@ Lukas, Leon},
     def get_hash(input_str) -> str:
         return hashlib.md5(input_str.encode("utf-8")).hexdigest()
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/deu/german_qu_ad_retrieval.py

@@ -58,7 +58,7 @@ class GermanQuADRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/ell/greek_civics_qa.py

@@ -31,7 +31,7 @@ class GreekCivicsQA(AbsTaskRetrieval):
         bibtex_citation="",
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
         # fetch both subsets of the dataset

mteb/tasks/retrieval/eng/bright_retrieval.py

@@ -83,7 +83,7 @@ def load_bright_data(
     return corpus, queries, relevant_docs
 
 
-def load_data(self, num_proc: int = 1, **kwargs) -> None:
+def load_data(self, num_proc: int | None = None, **kwargs) -> None:
     if self.data_loaded:
         return
 

mteb/tasks/retrieval/eng/chat_doctor_retrieval.py

@@ -36,7 +36,7 @@ class ChatDoctorRetrieval(AbsTaskRetrieval):
         },
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/fin_qa_retrieval.py

@@ -37,7 +37,7 @@ class FinQARetrieval(AbsTaskRetrieval):
         },
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/finance_bench_retrieval.py

@@ -37,7 +37,7 @@ class FinanceBenchRetrieval(AbsTaskRetrieval):
         },
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/hateful_memes_i2t_retrieval.py

@@ -90,7 +90,7 @@ class HatefulMemesI2TRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
         self.corpus, self.queries, self.relevant_docs = _load_data(

mteb/tasks/retrieval/eng/hateful_memes_t2i_retrieval.py

@@ -90,7 +90,7 @@ class HatefulMemesT2IRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
         self.corpus, self.queries, self.relevant_docs = _load_data(

mteb/tasks/retrieval/eng/hc3_finance_retrieval.py

@@ -37,7 +37,7 @@ class HC3FinanceRetrieval(AbsTaskRetrieval):
         },
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/lemb_narrative_qa_retrieval.py

@@ -55,7 +55,7 @@ Roark, Brian},
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/lemb_needle_retrieval.py

@@ -46,7 +46,7 @@ class LEMBNeedleRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/lemb_passkey_retrieval.py

@@ -46,7 +46,7 @@ class LEMBPasskeyRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/lemb_summ_screen_fd_retrieval.py

@@ -51,7 +51,7 @@ Villavicencio, Aline},
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/lemb_wikim_qa_retrieval.py

@@ -40,7 +40,7 @@ class LEMBWikimQARetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/lembqm_sum_retrieval.py

@@ -64,7 +64,7 @@ Zhou, Yichao},
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/lit_search_retrieval.py

@@ -35,7 +35,7 @@ class LitSearchRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
         self.corpus, self.queries, self.relevant_docs = {}, {}, {}

mteb/tasks/retrieval/eng/memotion_i2t_retrieval.py

@@ -113,7 +113,7 @@ class MemotionI2TRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
         self.corpus, self.queries, self.relevant_docs = _load_data(

mteb/tasks/retrieval/eng/memotion_t2i_retrieval.py

@@ -112,7 +112,7 @@ class MemotionT2IRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
         self.corpus, self.queries, self.relevant_docs = _load_data(

mteb/tasks/retrieval/eng/ml_questions.py

@@ -55,7 +55,7 @@ Reddy, Siva},
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
         self.corpus, self.queries, self.relevant_docs = {}, {}, {}

mteb/tasks/retrieval/eng/nano_argu_ana_retrieval.py

@@ -40,7 +40,7 @@ class NanoArguAnaRetrieval(AbsTaskRetrieval):
         adapted_from=["ArguAna"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/nano_climate_fever_retrieval.py

@@ -44,7 +44,7 @@ class NanoClimateFeverRetrieval(AbsTaskRetrieval):
         adapted_from=["ClimateFEVER"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/nano_db_pedia_retrieval.py

@@ -42,7 +42,7 @@ class NanoDBPediaRetrieval(AbsTaskRetrieval):
         adapted_from=["DBPedia"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/nano_fever_retrieval.py

@@ -57,7 +57,7 @@ Stent, Amanda},
         adapted_from=["FEVER"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/nano_fi_qa2018_retrieval.py

@@ -43,7 +43,7 @@ class NanoFiQA2018Retrieval(AbsTaskRetrieval):
         adapted_from=["FiQA2018"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/nano_hotpot_qa_retrieval.py

@@ -60,7 +60,7 @@ Tsujii, Jun{'}ichi},
         adapted_from=["HotpotQA"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/nano_msmarco_retrieval.py

@@ -55,7 +55,7 @@ Li Deng},
         adapted_from=["MSMARCO"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/nano_nf_corpus_retrieval.py

@@ -46,7 +46,7 @@ class NanoNFCorpusRetrieval(AbsTaskRetrieval):
         adapted_from=["NFCorpus"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/nano_nq_retrieval.py

@@ -46,7 +46,7 @@ Linguistics},
         adapted_from=["NQ"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/nano_quora_retrieval.py

@@ -45,7 +45,7 @@ class NanoQuoraRetrieval(AbsTaskRetrieval):
         adapted_from=["QuoraRetrieval"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/nano_sci_fact_retrieval.py

@@ -42,7 +42,7 @@ class NanoSciFactRetrieval(AbsTaskRetrieval):
         adapted_from=["SciFact"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/nano_scidocs_retrieval.py

@@ -44,7 +44,7 @@ class NanoSCIDOCSRetrieval(AbsTaskRetrieval):
         adapted_from=["SCIDOCS"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/nano_touche2020_retrieval.py

@@ -53,7 +53,7 @@ Questions}},
         adapted_from=["Touche2020"],
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/narrative_qa_retrieval.py

@@ -43,7 +43,7 @@ class NarrativeQARetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/r2_med_retrieval.py

@@ -70,7 +70,7 @@ class R2MEDBiologyRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 
@@ -114,7 +114,7 @@ class R2MEDBioinformaticsRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 
@@ -158,7 +158,7 @@ class R2MEDMedicalSciencesRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 
@@ -202,7 +202,7 @@ class R2MEDMedXpertQAExamRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 
@@ -246,7 +246,7 @@ class R2MEDMedQADiagRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 
@@ -290,7 +290,7 @@ class R2MEDPMCTreatmentRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 
@@ -334,7 +334,7 @@ class R2MEDPMCClinicalRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 
@@ -378,7 +378,7 @@ class R2MEDIIYiClinicalRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
 

mteb/tasks/retrieval/eng/sci_mmir_i2t_retrieval.py

@@ -88,7 +88,7 @@ class SciMMIRI2TRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
         self.corpus, self.queries, self.relevant_docs = _load_data(

mteb/tasks/retrieval/eng/sci_mmir_t2i_retrieval.py

@@ -88,7 +88,7 @@ class SciMMIRT2IRetrieval(AbsTaskRetrieval):
 """,
     )
 
-    def load_data(self, num_proc: int = 1, **kwargs) -> None:
+    def load_data(self, num_proc: int | None = None, **kwargs) -> None:
         if self.data_loaded:
             return
         self.corpus, self.queries, self.relevant_docs = _load_data(