diff --git a/tests/test_pipeline.py b/tests/test_pipeline.py index 0f1c2c5dd..957634183 100644 --- a/tests/test_pipeline.py +++ b/tests/test_pipeline.py @@ -92,7 +92,7 @@ def test_validate_steps(self): ("bad", Thing()), ("model", MockEstimator()), ] - ) + )._validate_steps() # validate a bad intermediate transformer on the VisualPipeline with pytest.raises(TypeError): @@ -102,15 +102,17 @@ def test_validate_steps(self): ("bad", Thing()), ("model", MockEstimator()), ] - ) + )._validate_steps() # validate a bad final estimator on the Pipeline with pytest.raises(TypeError): - Pipeline([("real", MockTransformer()), ("bad", Thing())]) + Pipeline([("real", MockTransformer()), ("bad", Thing())])._validate_steps() # validate a bad final estimator on the VisualPipeline with pytest.raises(TypeError): - VisualPipeline([("real", MockTransformer()), ("bad", Thing())]) + VisualPipeline( + [("real", MockTransformer()), ("bad", Thing())] + )._validate_steps() # validate visual transformers on a Pipeline try: