diff --git a/newrelic/hooks/mlmodel_sklearn.py b/newrelic/hooks/mlmodel_sklearn.py index 539d6ddb7..8d49b92a9 100644 --- a/newrelic/hooks/mlmodel_sklearn.py +++ b/newrelic/hooks/mlmodel_sklearn.py @@ -120,12 +120,10 @@ def wrap_metric_scorer(wrapped, instance, args, kwargs): if not isinstance(score, (str, int, float, bool)): if hasattr(score, "__iter__"): for i, s in enumerate(score): - # ModelName/TrainingStep//accuracy_score[] transaction._add_agent_attribute( "%s/TrainingStep/%s/%s[%s]" % (model_name, training_step, wrapped.__name__, i), s ) else: - # ModelName/TrainingStep//accuracy_score transaction._add_agent_attribute("%s/TrainingStep/%s/%s" % (model_name, training_step, wrapped.__name__), score) return score