From 8d38e268babb218ad08c3d0b5d5f4c73a1ef4d19 Mon Sep 17 00:00:00 2001 From: Uma Annamalai Date: Thu, 14 Sep 2023 11:04:03 -0700 Subject: [PATCH] Add wrap_mlmodel to newrelic.agent (#920) --- newrelic/agent.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/newrelic/agent.py b/newrelic/agent.py index 70dd3a14d..2c7f0fb85 100644 --- a/newrelic/agent.py +++ b/newrelic/agent.py @@ -153,6 +153,7 @@ def __asgi_application(*args, **kwargs): from newrelic.api.message_transaction import ( wrap_message_transaction as __wrap_message_transaction, ) +from newrelic.api.ml_model import wrap_mlmodel as __wrap_mlmodel from newrelic.api.profile_trace import ProfileTraceWrapper as __ProfileTraceWrapper from newrelic.api.profile_trace import profile_trace as __profile_trace from newrelic.api.profile_trace import wrap_profile_trace as __wrap_profile_trace @@ -207,11 +208,6 @@ def __asgi_application(*args, **kwargs): # EXPERIMENTAL - Generator traces are currently experimental and may not # exist in this form in future versions of the agent. - -# EXPERIMENTAL - Profile traces are currently experimental and may not -# exist in this form in future versions of the agent. - - initialize = __initialize extra_settings = __wrap_api_call(__extra_settings, "extra_settings") global_settings = __wrap_api_call(__global_settings, "global_settings") @@ -343,3 +339,4 @@ def __asgi_application(*args, **kwargs): wrap_out_function = __wrap_api_call(__wrap_out_function, "wrap_out_function") insert_html_snippet = __wrap_api_call(__insert_html_snippet, "insert_html_snippet") verify_body_exists = __wrap_api_call(__verify_body_exists, "verify_body_exists") +wrap_mlmodel = __wrap_api_call(__wrap_mlmodel, "wrap_mlmodel")