Cleaner exit

allenai · Nov 15, 2024 · 2c52664 · 2c52664
1 parent 77c82fd
commit 2c52664
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pdelfin/beakerpipeline.py b/pdelfin/beakerpipeline.py
@@ -785,10 +785,11 @@ async def main():
     await asyncio.gather(*worker_tasks, return_exceptions=True)
 
     # Wait for server to stop
+    process_pool.shutdown(wait=False)
+
     sglang_server.cancel()
     metrics_task.cancel()
-    logger.info("Work done, force exitting...")
-    sys.exit(0)
+    logger.info("Work done")
 
 if __name__ == "__main__":
     asyncio.run(main(), debug=True)

diff --git a/pdelfin/version.py b/pdelfin/version.py
@@ -2,7 +2,7 @@
 _MINOR = "1"
 # On main and in a nightly release the patch should be one ahead of the last
 # released build.
-_PATCH = "18"
+_PATCH = "19"
 # This is mainly for nightly builds which have the suffix ".dev$DATE". See
 # https://semver.org/#is-v123-a-semantic-version for the semantics.
 _SUFFIX = ""