Fix failing tests

pydantic · Dec 13, 2024 · 17cfbc7 · 17cfbc7
1 parent fb7789c
commit 17cfbc7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/logfire/_internal/main.py b/logfire/_internal/main.py
@@ -1921,7 +1921,7 @@ def __exit__(self, exc_type: type[BaseException] | None, exc_value: BaseExceptio
 
         if self._span is not None:
             span_context = self._span.get_span_context()
-            del OPEN_LOGFIRE_SPANS_BY_ID[(span_context.trace_id, span_context.span_id)]
+            OPEN_LOGFIRE_SPANS_BY_ID.pop((span_context.trace_id, span_context.span_id), None)
         OPEN_SPANS.remove(self)
 
         context_api.detach(self._token)