From 4ffb6036733217acaffbc35593c9cbb9c4ac18ff Mon Sep 17 00:00:00 2001 From: Uma Annamalai Date: Tue, 21 May 2024 14:33:44 -0700 Subject: [PATCH] Fix sklearn ensemble tests. --- tests/mlmodel_sklearn/test_ensemble_models.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/mlmodel_sklearn/test_ensemble_models.py b/tests/mlmodel_sklearn/test_ensemble_models.py index 8a4a2f8e2e..69c10f1b28 100644 --- a/tests/mlmodel_sklearn/test_ensemble_models.py +++ b/tests/mlmodel_sklearn/test_ensemble_models.py @@ -232,7 +232,6 @@ def test_above_v1_1_model_methods_wrapped_in_function_trace(ensemble_model_name, ("Function/MLModel/Sklearn/Named/HistGradientBoostingClassifier.fit", 1), ("Function/MLModel/Sklearn/Named/HistGradientBoostingClassifier.predict", 2), ("Function/MLModel/Sklearn/Named/HistGradientBoostingClassifier.score", 1), - ("Function/MLModel/Sklearn/Named/HistGradientBoostingClassifier.predict_proba", 3), ], "HistGradientBoostingRegressor": [ ("Function/MLModel/Sklearn/Named/HistGradientBoostingRegressor.fit", 1),