From 1f5934e4b8176f00f87e579ebcf7c6fc2c230d7f Mon Sep 17 00:00:00 2001 From: Deepyaman Datta Date: Mon, 16 Sep 2024 10:00:55 -0600 Subject: [PATCH] test(core): check that `Pipeline.set_params` works --- tests/test_core.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_core.py b/tests/test_core.py index 4592971..70f0f78 100644 --- a/tests/test_core.py +++ b/tests/test_core.py @@ -209,7 +209,8 @@ def test_can_use_in_sklearn_pipeline(): # get/set params works params = p.get_params() - p.set_params(**params) + p.set_params(**params | {"recipe__scalestandard__inputs": ml.numeric()}) + assert p["recipe"].steps[1].inputs == ml.numeric() # fit and predict work p.fit(X, y)