test(core): check that Pipeline.set_params works

ibis-project · Sep 16, 2024 · 1f5934e · 1f5934e
1 parent 6a1ec0a
commit 1f5934e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)