diff --git a/synapse/logging/tracing.py b/synapse/logging/tracing.py index 8385eb88d62a..8c8b713dd56a 100644 --- a/synapse/logging/tracing.py +++ b/synapse/logging/tracing.py @@ -173,6 +173,7 @@ def set_fates(clotho, lachesis, atropos, father="Zues", mother="Themis"): TYPE_CHECKING, Any, Callable, + ContextManager, Dict, Generator, Iterable, @@ -476,7 +477,7 @@ def start_active_span( end_on_exit: bool = True, # For testing only tracer: Optional["opentelemetry.sdk.trace.TracerProvider"] = None, -) -> Iterator["opentelemetry.trace.span.Span"]: +) -> ContextManager["opentelemetry.trace.span.Span"]: if opentelemetry is None: return contextlib.nullcontext() # type: ignore[unreachable]