From eec573465c52ba4b15206c1927bc51e93bb18ef4 Mon Sep 17 00:00:00 2001 From: antoinedemathelin <45264923+antoinedemathelin@users.noreply.github.com> Date: Thu, 30 Sep 2021 08:20:06 +0200 Subject: [PATCH] Update test_utils.py --- tests/test_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_utils.py b/tests/test_utils.py index cabe93f..ec2ab17 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -270,7 +270,7 @@ def test_check_network_compile(): Ridge(alpha=10, fit_intercept=False, tol=0.1), DecisionTreeClassifier(max_depth=10), AdaBoostRegressor(Ridge(alpha=0.01)), - TransformedTargetRegressor(Ridge(alpha=25), StandardScaler()), + TransformedTargetRegressor(regressor=Ridge(alpha=25), transformer=StandardScaler()), MultiOutputRegressor(Ridge(alpha=0.3)), make_pipeline(StandardScaler(), Ridge(alpha=0.2)), KerasClassifier(_get_model_Sequential, input_shape=(1,)), @@ -458,4 +458,4 @@ def test_make_regression_da(): assert Xs.shape == (1000, 10) assert len(ys) == 1000 assert Xt.shape == (1000, 10) - assert len(yt) == 1000 \ No newline at end of file + assert len(yt) == 1000