Merge pull request #165 from siapy/fix

fix: Update n_jobs default value to use all processors

siapy/optimizers/configs.py

@@ -29,7 +29,7 @@ class OptimizeStudyConfig(BaseModel):
     model_config = ConfigDict(arbitrary_types_allowed=True)
     n_trials: int | None = None
     timeout: float | None = None
-    n_jobs: int = 1
+    n_jobs: int = -1
     catch: Iterable[type[Exception]] | type[Exception] = ()
     callbacks: (
         list[Callable[[optuna.study.Study, optuna.trial.FrozenTrial], None]] | None

siapy/optimizers/scorers.py

@@ -1,5 +1,5 @@
 from functools import partial
-from typing import Iterable, Literal
+from typing import Annotated, Iterable, Literal
 
 import numpy as np
 from sklearn import model_selection
@@ -38,6 +38,10 @@ def init_cross_validator_scorer(
         | Iterable
         | Literal["RepeatedKFold", "RepeatedStratifiedKFold"]
         | None = None,
+        n_jobs: Annotated[
+            int | None,
+            "Number of jobs to run in parallel. `-1` means using all processors.",
+        ] = None,
     ) -> "Scorer":
         if isinstance(cv, str) and cv in ["RepeatedKFold", "RepeatedStratifiedKFold"]:
             cv = initialize_object(
@@ -52,7 +56,7 @@ def init_cross_validator_scorer(
             scoring=scoring,
             cv=cv,  # type: ignore
             groups=None,
-            n_jobs=1,
+            n_jobs=n_jobs,
             verbose=0,
             fit_params=None,
             pre_dispatch=1,

tests/optimizers/test_optimizers_configs.py

@@ -24,7 +24,7 @@ def test_optimize_study_config_defaults():
     config = OptimizeStudyConfig()
     assert config.n_trials is None
     assert config.timeout is None
-    assert config.n_jobs == 1
+    assert config.n_jobs == -1
     assert config.catch == ()
     assert config.callbacks is None
     assert config.gc_after_trial is False