Merge pull request #1148 from newrelic/fix-ensemble-sklearn-tests

Fix sklearn ensemble tests.
newrelic · May 23, 2024 · 20c5298 · 20c5298
2 parents ac28232 + 08a3971
commit 20c5298
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),