From ac64e9f0ebb31b08ad8b36c2fa374698dd4ac26f Mon Sep 17 00:00:00 2001
From: Quentin Pradet <quentin.pradet@elastic.co>
Date: Tue, 10 Sep 2024 11:24:01 +0400
Subject: [PATCH] Fix _otel lint (#2654)

---
 elasticsearch/_otel.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/elasticsearch/_otel.py b/elasticsearch/_otel.py
index 49de9201c..264d8aa35 100644
--- a/elasticsearch/_otel.py
+++ b/elasticsearch/_otel.py
@@ -98,7 +98,7 @@ 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]:
@@ -106,5 +106,5 @@ def use_span(self, span: OpenTelemetrySpan) -> Generator[None, None, None]:
             yield
             return
 
-        with trace.use_span(span):
+        with trace.use_span(span.otel_span):
             yield