fix: Store before_shutdown hook in def

src/axiom_py/logging.py

@@ -34,7 +34,11 @@ def __init__(self, client: Client, dataset: str, level=NOTSET, interval=1):
         self.timer = Timer(self.interval, self.flush)
 
         # Make sure we flush before the client shuts down
-        client.before_shutdown(lambda: self.flush(), self.timer.cancel())
+        def before_shutdown():
+            self.flush()
+            self.timer.cancel()
+
+        client.before_shutdown(before_shutdown)
 
     def emit(self, record):
         """Emit sends a log to Axiom."""