diff --git a/newrelic/api/datastore_trace.py b/newrelic/api/datastore_trace.py index d3a57ff92a..1561293f64 100644 --- a/newrelic/api/datastore_trace.py +++ b/newrelic/api/datastore_trace.py @@ -287,7 +287,9 @@ def datastore_trace(product, target, operation, host=None, port_path_or_id=None, ) -def wrap_datastore_trace(module, object_path, product, target, operation, host=None, port_path_or_id=None, database_name=None): +def wrap_datastore_trace( + module, object_path, product, target, operation, host=None, port_path_or_id=None, database_name=None +): """Method applies custom timing to datastore query. :param module: Module containing the method to be instrumented.