type hints

tests/test_utils/test_comparison.py

@@ -19,7 +19,7 @@ def test_are_equal(self) -> None:
             get_standardization_pipeline,
             get_morgan_physchem_rf_pipeline,
         ]
-        for pipeline_method in pipline_method_list:
+        for pipeline_method in pipline_method_list:  # type: Callable[[int], Pipeline]
             pipeline_a = pipeline_method()
             pipeline_b = pipeline_method()
             self.assertTrue(check_pipelines_equivalent(pipeline_a, pipeline_b))

tests/utils/default_models.py

@@ -25,9 +25,14 @@
 from molpipeline.post_prediction import PostPredictionWrapper
 
 
-def get_morgan_physchem_rf_pipeline() -> Pipeline:
+def get_morgan_physchem_rf_pipeline(n_jobs: int = 1) -> Pipeline:
     """Get a pipeline combining Morgan fingerprints and physicochemical properties with a RandomForestClassifier.
 
+    Parameters
+    ----------
+    n_jobs: int, default=-1
+        Number of parallel jobs to use.
+
     Returns
     -------
     Pipeline
@@ -47,20 +52,20 @@ def get_morgan_physchem_rf_pipeline() -> Pipeline:
                 ),
             ),
             ("error_filter", error_filter),
-            ("rf", RandomForestClassifier()),
+            ("rf", RandomForestClassifier(n_jobs=n_jobs)),
             (
                 "filter_reinserter",
                 PostPredictionWrapper(
                     FilterReinserter.from_error_filter(error_filter, None)
                 ),
             ),
         ],
-        n_jobs=1,
+        n_jobs=n_jobs,
     )
     return pipeline
 
 
-def get_standardization_pipeline(n_jobs: int = -1) -> Pipeline:
+def get_standardization_pipeline(n_jobs: int = 1) -> Pipeline:
     """Get the standardization pipeline.
 
     Parameters