kill process when the worker crashes, increase batch limit1.5.0

1 files changed, 4 insertions, 6 deletions

diff --git a/api.py b/api.py
index b75beb9..b1321d5 100644
--- a/api.py
+++ b/api.py
@@ -80,15 +80,13 @@ class Compiler(joblib.worker.Worker):
 
 
 async def startup():
-    for _ in range(1):
-        worker = Compiler()
-        await worker.connect(db_config, queue_db)
-        await worker.setup()
-        await worker.start(batchlimit=750)
+    worker = Compiler()
+    await worker.connect(db_config, queue_db)
+    await worker.setup()
+    await worker.run(batchlimit=1000)
 
 
 logging.basicConfig(level=logging.DEBUG)
 
 loop = asyncio.get_event_loop()
 loop.run_until_complete(startup())
-loop.run_forever()