From 29c1ef0b8faebcf014a35d4858bcb31b21674790 Mon Sep 17 00:00:00 2001 From: Munir Abdinur Date: Tue, 4 Feb 2025 20:10:31 -0500 Subject: [PATCH] fix fmt --- tests/tracer/test_memory_leak.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/tests/tracer/test_memory_leak.py b/tests/tracer/test_memory_leak.py index 2963268fd65..b13cc0ec4ee 100644 --- a/tests/tracer/test_memory_leak.py +++ b/tests/tracer/test_memory_leak.py @@ -1,26 +1,21 @@ """ Variety of test cases ensuring that ddtrace does not leak memory. """ -from typing import TYPE_CHECKING from weakref import WeakValueDictionary import pytest +from ddtrace.trace import Span +from ddtrace.trace import Tracer from tests.utils import DummyTracer -if TYPE_CHECKING: # pragma: no cover - from ddtrace.trace import Span # noqa:F401 - from ddtrace.trace import Tracer # noqa:F401 - - @pytest.fixture def tracer() -> DummyTracer: return DummyTracer() -def trace(weakdict: WeakValueDictionary, tracer: Tracer, *args, **kwargs): - # type: (...) -> Span +def trace(weakdict: WeakValueDictionary, tracer: Tracer, *args, **kwargs) -> Span: """Return a span created from ``tracer`` and add it to the given weak dictionary.