datadog: fix start_as_current_span

TheShellLand · Apr 13, 2024 · 917953e · 917953e
1 parent 0b24fe2
commit 917953e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/automon/integrations/openTelemetryWrapper/client.py b/automon/integrations/openTelemetryWrapper/client.py
@@ -49,7 +49,7 @@ async def record_exception(self, exception: Exception):
 
     def start_as_current_span(
             self, name: str,
-            attributes: types.Attributes = None,
+            attributes: any = None,
             **kwargs
     ):
         logger.debug(dict(name=name, attributes=attributes, kwargs=kwargs))
@@ -71,7 +71,7 @@ async def start_producer(self):
         return
 
     async def test(self):
-        with self.config.tracer.start_as_current_span(name='rootSpan') as trace_root:
+        with self.start_as_current_span(name='rootSpan') as trace_root:
             self.add_event('AAAAAAAA')
 
             with self.config.tracer.start_as_current_span(name='childSpan') as trace_child: