fix n_jobs on spa preset

hackingmaterials · Jul 28, 2020 · c5a66e3 · c5a66e3
1 parent 64767f4
commit c5a66e3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/automatminer/presets.py b/automatminer/presets.py
@@ -85,7 +85,8 @@ def get_preset_config(preset: str = "express", **powerups) -> dict:
             "cleaner": DataCleaner(),
         }
     elif preset == "express_single":
-        rf_args = {"n_estimators": 500, "max_depth": 5, "n_jobs": n_jobs_kwargs}
+        rf_args = {"n_estimators": 500, "max_depth": 5}
+        rf_args.update(n_jobs_kwargs)
         config = {
             "learner": SinglePipelineAdaptor(
                 regressor=RandomForestRegressor(**rf_args),