fix lint

Author: sohankshirsagar

drift/core/mode_utils.py

@@ -8,12 +8,13 @@
 from __future__ import annotations
 
 import logging
-from typing import TYPE_CHECKING, Callable, TypeVar
+from collections.abc import Callable
+from typing import TYPE_CHECKING, TypeVar
 
 from opentelemetry.trace import SpanKind as OTelSpanKind
 
 if TYPE_CHECKING:
-    from .tracing.span_utils import SpanInfo
+    pass
 
 logger = logging.getLogger(__name__)
 

drift/core/tracing/td_span_processor.py

@@ -13,7 +13,6 @@
 from opentelemetry.sdk.trace import ReadableSpan, SpanProcessor
 from opentelemetry.trace import Span
 
-from ..sampling import should_sample
 from ..trace_blocking_manager import TraceBlockingManager, should_block_span
 from ..types import TD_INSTRUMENTATION_LIBRARY_NAME, TuskDriftMode, replay_trace_id_context
 from ..types import SpanKind as TdSpanKind

drift/instrumentation/django/middleware.py

@@ -13,7 +13,9 @@
 
 if TYPE_CHECKING:
     from django.http import HttpRequest, HttpResponse  # type: ignore[import-not-found]
+from ...core.mode_utils import handle_record_mode
 from ...core.tracing import TdSpanAttributes
+from ...core.tracing.span_utils import CreateSpanOptions, SpanInfo, SpanUtils
 from ...core.types import (
     CleanSpanData,
     Duration,
@@ -33,8 +35,6 @@
     build_output_schema_merges,
     build_output_value,
 )
-from ...core.mode_utils import handle_record_mode
-from ...core.tracing.span_utils import CreateSpanOptions, SpanInfo, SpanUtils
 
 
 class DriftMiddleware:

drift/instrumentation/fastapi/instrumentation.py

@@ -18,16 +18,15 @@
     Receive = Callable[[], Awaitable[dict[str, Any]]]
     Send = Callable[[dict[str, Any]], Awaitable[None]]
 
-from opentelemetry import trace
 from opentelemetry.trace import SpanKind as OTelSpanKind
 from opentelemetry.trace import Status
 from opentelemetry.trace import StatusCode as OTelStatusCode
 
 from ...core.drift_sdk import TuskDrift
-from ...core.mode_utils import handle_record_mode
-from ...core.tracing.span_utils import CreateSpanOptions, SpanInfo, SpanUtils
 from ...core.json_schema_helper import JsonSchemaHelper, SchemaMerge
+from ...core.mode_utils import handle_record_mode
 from ...core.tracing import TdSpanAttributes
+from ...core.tracing.span_utils import CreateSpanOptions, SpanInfo, SpanUtils
 from ...core.types import (
     PackageType,
     SpanKind,

drift/instrumentation/wsgi/handler.py

@@ -14,7 +14,7 @@
 
 from opentelemetry import context as otel_context
 from opentelemetry.trace import SpanKind as OTelSpanKind
-from opentelemetry.trace import Status, set_span_in_context
+from opentelemetry.trace import Status
 from opentelemetry.trace import StatusCode as OTelStatusCode
 
 logger = logging.getLogger(__name__)