formatting

hackingmaterials · Nov 12, 2019 · 9f0cb61 · 9f0cb61
1 parent 2340e0f
commit 9f0cb61
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/automatminer/presets.py b/automatminer/presets.py
@@ -85,12 +85,7 @@ def get_preset_config(preset: str = "express", **powerups) -> dict:
             "cleaner": DataCleaner(),
         }
     elif preset == "express_single":
-        nj = n_jobs_kwargs["n_jobs"]
-        xgb_kwargs = {
-            "n_estimators": 300,
-            "max_depth": 3,
-            "n_jobs": n_jobs_kwargs,
-        }
+        xgb_kwargs = {"n_estimators": 300, "max_depth": 3, "n_jobs": n_jobs_kwargs}
         config = {
             "learner": SinglePipelineAdaptor(
                 regressor=XGBRegressor(**xgb_kwargs),