From 4d7b5e061c6bfde9dd18fcacf2ced0572ed1b7e8 Mon Sep 17 00:00:00 2001 From: Imran Ahamed Date: Fri, 29 May 2026 15:51:27 -0500 Subject: [PATCH] =?UTF-8?q?fix:=20LRL1/LRL2=20=E2=80=94=20migrate=20off=20?= =?UTF-8?q?sklearn=201.10=20deprecations=20(penalty,=20n=5Fjobs)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- flaml/automl/model.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/flaml/automl/model.py b/flaml/automl/model.py index af2e249437..abf4c95c98 100644 --- a/flaml/automl/model.py +++ b/flaml/automl/model.py @@ -2034,7 +2034,11 @@ def config2params(self, config: dict) -> dict: params = super().config2params(config) params["tol"] = params.get("tol", 0.0001) params["solver"] = params.get("solver", "saga") - params["penalty"] = params.get("penalty", "l1") + if SKLEARN_VERSION >= "1.8": + params["l1_ratio"] = params.get("l1_ratio", 1.0) + params.pop("n_jobs", None) + else: + params["penalty"] = params.get("penalty", "l1") return params def __init__(self, task="binary", **config): @@ -2063,7 +2067,10 @@ def config2params(self, config: dict) -> dict: params = super().config2params(config) params["tol"] = params.get("tol", 0.0001) params["solver"] = params.get("solver", "lbfgs") - params["penalty"] = params.get("penalty", "l2") + if SKLEARN_VERSION >= "1.8": + params.pop("n_jobs", None) + else: + params["penalty"] = params.get("penalty", "l2") return params def __init__(self, task="binary", **config):