From 80401a5a2d6268711582b7fe31b1e644a59b35ab Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Mon, 29 Dec 2025 01:30:06 +0100 Subject: [PATCH] refactor(quickadapter): silence warning at HPO with histgb MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- quickadapter/user_data/strategies/Utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/quickadapter/user_data/strategies/Utils.py b/quickadapter/user_data/strategies/Utils.py index 6b958e8..c3b28cb 100644 --- a/quickadapter/user_data/strategies/Utils.py +++ b/quickadapter/user_data/strategies/Utils.py @@ -2072,7 +2072,7 @@ def fit_regressor( Args: regressor: Type of regressor. model_training_parameters: Copied internally to avoid side effects. - callbacks: Additional callbacks (pruning callbacks added automatically when trial is set). + callbacks: Additional callbacks. trial: Optuna trial for pruning and random state offset. """ model_training_parameters = model_training_parameters.copy() @@ -2216,7 +2216,7 @@ def fit_regressor( model = pruning_callback( model=model, X=X, - y=y, + y=y.to_numpy().ravel(), X_val=X_val, y_val=y_val, sample_weight=train_weights, -- 2.53.0