chore: Dont import Span from deprecated module

kolonialno · Jun 27, 2024 · 342e82e · 342e82e
1 parent 389ff5f
commit 342e82e
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 27 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/troncos/tracing/_span.py b/troncos/tracing/_span.py
@@ -1,7 +1,6 @@
 from typing import Any
 
-from ddtrace import constants, ext
-from ddtrace.span import Span as DDSpan
+from ddtrace import Span as DDSpan, constants, ext
 from opentelemetry.attributes import BoundedAttributes  # type: ignore
 from opentelemetry.sdk.resources import SERVICE_NAME, Resource
 from opentelemetry.sdk.trace import (

diff --git a/troncos/tracing/_writer.py b/troncos/tracing/_writer.py
@@ -1,7 +1,7 @@
 from typing import Any
 
+from ddtrace import Span
 from ddtrace.internal.writer.writer import TraceWriter
-from ddtrace.span import Span
 from opentelemetry.sdk.resources import Resource
 
 from ._exporter import Exporter

diff --git a/troncos/tracing/decorators.py b/troncos/tracing/decorators.py
@@ -8,7 +8,6 @@
 from typing import Awaitable, Callable, ParamSpec, Type, TypeVar, cast, overload
 
 import ddtrace
-from ddtrace.span import Span
 
 _TRACE_IGNORE_ATTR = "_trace_ignore"
 
@@ -26,7 +25,7 @@ def trace_block(
     service: str | None = None,
     span_type: str | None = None,
     attributes: dict[str, str] | None = None,
-) -> Generator[Span, None, None]:
+) -> Generator[ddtrace.Span, None, None]:
     """
     Trace a code block using a with statement. Example: