Merge pull request #8 from adapt-python/antoinedemathelin-patch-2

test: Fix bug with TransformedTargetRegressor
adapt-python · Sep 30, 2021 · 5dd2483 · 5dd2483
2 parents 926635a + eec5734
commit 5dd2483
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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
+    assert len(yt) == 1000