Skip to content

Commit 6d68a6a

Browse files
fix: api-server worker init
1 parent 4275912 commit 6d68a6a

File tree

3 files changed

+5
-6
lines changed
  • packages/celery-library/src/celery_library
  • services

3 files changed

+5
-6
lines changed

packages/celery-library/src/celery_library/signals.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@
1313

1414

1515
def on_worker_init(
16-
app_server: BaseAppServer,
1716
sender: WorkController,
17+
app_server: BaseAppServer,
1818
**_kwargs,
1919
) -> None:
2020
startup_complete_event = threading.Event()

services/api-server/src/simcore_service_api_server/celery_worker/worker_main.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,4 @@ def worker_init_wrapper(sender, **_kwargs):
3737
assert _settings.API_SERVER_CELERY # nosec
3838
app_server = FastAPIAppServer(app=create_app(_settings))
3939

40-
return partial(on_worker_init, app_server, _settings.API_SERVER_CELERY)(
41-
sender, **_kwargs
42-
)
40+
return partial(on_worker_init, app_server=app_server)(sender, **_kwargs)

services/storage/src/simcore_service_storage/modules/celery/worker_main.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Main application to be deployed in for example uvicorn."""
2+
23
from celery.signals import worker_init, worker_shutdown # type: ignore[import-untyped]
34
from celery_library.common import create_app as create_celery_app
45
from celery_library.signals import (
@@ -29,8 +30,8 @@
2930
app_server = FastAPIAppServer(app=create_app(_settings))
3031

3132

32-
def worker_init_wrapper(sender, **_kwargs):
33-
return on_worker_init(sender, app_server, **_kwargs)
33+
def worker_init_wrapper(sender, **kwargs):
34+
return on_worker_init(sender, app_server, **kwargs)
3435

3536

3637
worker_init.connect(worker_init_wrapper)

0 commit comments

Comments
 (0)