remove instrument

truera · Dec 19, 2024 · 27bcac5 · 27bcac5
1 parent 049b404
commit 27bcac5
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/core/trulens/experimental/otel_tracing/core/instrument.py b/src/core/trulens/experimental/otel_tracing/core/instrument.py
@@ -3,7 +3,6 @@
 from typing import Any, Callable, Optional, Union
 
 from opentelemetry import trace
-from trulens.apps.custom import instrument as custom_instrument
 from trulens.experimental.otel_tracing.core.init import TRULENS_SERVICE_NAME
 from trulens.experimental.otel_tracing.core.semantic import (
     TRULENS_SELECTOR_NAME,
@@ -46,25 +45,22 @@ def wrapper(*args, **kwargs):
                 )
             ) as parent_span:
                 span = trace.get_current_span()
-                _instrumented_object, *rest_args = args
 
                 span.set_attribute("name", func.__name__)
                 span.set_attribute("kind", "SPAN_KIND_TRULENS")
                 span.set_attribute(
                     "parent_span_id", parent_span.get_span_context().span_id
                 )
 
-                ret = custom_instrument(func)(*args, **kwargs)
+                ret = func(*args, **kwargs)
 
                 try:
                     attributes_to_add = {}
 
                     # Set the user provider attributes.
                     if attributes:
                         if callable(attributes):
-                            attributes_to_add = attributes(
-                                ret, *rest_args, **kwargs
-                            )
+                            attributes_to_add = attributes(ret, *args, **kwargs)
                         else:
                             attributes_to_add = attributes