Fix sklearn ensemble tests.

newrelic · May 21, 2024 · 4ffb603 · 4ffb603
1 parent 6d30f32
commit 4ffb603
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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),