Specify error code in WorkerShutdown exception

GooeyAI · Aug 12, 2024 · d775038 · d775038
1 parent 6829a49
commit d775038
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/celeryconfig.py b/celeryconfig.py
@@ -43,7 +43,7 @@ def init(**kwargs):
             # for some reason, celery seems to swallow exceptions in init
             print(f"Error loading {model_id}:")
             traceback.print_exc()
-            raise WorkerShutdown()
+            raise WorkerShutdown(1)
 
     init_fns.append(init)