fix: Overload __exit__ method for NoOpTracer

Aleph-Alpha · Aug 12, 2024 · d63746a · d63746a
1 parent 73db2da
commit d63746a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/intelligence_layer/core/tracer/tracer.py b/src/intelligence_layer/core/tracer/tracer.py
@@ -376,6 +376,14 @@ def end(self, timestamp: Optional[datetime] = None) -> None:
     def export_for_viewing(self) -> Sequence[ExportedSpan]:
         return []
 
+    def __exit__(
+            self,
+            exc_type: Optional[type[BaseException]],
+            exc_value: Optional[BaseException],
+            _traceback: Optional[TracebackType],
+    ) -> None:
+        pass
+
 
 class JsonSerializer(RootModel[PydanticSerializable]):
     root: SerializeAsAny[PydanticSerializable]