Fix _otel lint (#2654)

elastic · Sep 10, 2024 · ac64e9f · ac64e9f
1 parent b8633ba
commit ac64e9f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/elasticsearch/_otel.py b/elasticsearch/_otel.py
@@ -98,13 +98,13 @@ def helpers_span(self, span_name: str) -> Generator[OpenTelemetrySpan, None, Non
             otel_span.set_attribute("db.operation", span_name)
             # Without a request method, Elastic APM does not display the traces
             otel_span.set_attribute("http.request.method", "null")
-            yield otel_span
+            yield OpenTelemetrySpan(otel_span)
 
     @contextlib.contextmanager
     def use_span(self, span: OpenTelemetrySpan) -> Generator[None, None, None]:
         if not self.enabled or self.tracer is None:
             yield
             return
 
-        with trace.use_span(span):
+        with trace.use_span(span.otel_span):
             yield