catch keyboard interrupt in logging process

apoorvkh · apoorvkh · commit bb1de1cc540e · 2024-11-01T22:27:50.000-04:00
diff --git a/pixi.lock b/pixi.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "torchrunx"
-version = "0.2.2"
+version = "0.2.3"
 authors = [
     {name = "Apoorv Khandelwal", email = "mail@apoorvkh.com"},
     {name = "Peter Curtin", email = "peter_curtin@brown.edu"},
diff --git a/src/torchrunx/utils/logging.py b/src/torchrunx/utils/logging.py
@@ -231,7 +231,10 @@ def start_logging_server(
         handlers=log_handlers,
     )
 
-    log_receiver.serve_forever()
+    try:
+        log_receiver.serve_forever()
+    except KeyboardInterrupt:
+        sys.exit(0)
 
     while not stop_event.is_set():
         pass
