File tree Expand file tree Collapse file tree 1 file changed +2
-6
lines changed
services/api-server/tests/unit/api_functions/celery Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Original file line number Diff line number Diff line change 66
77import datetime
88from collections .abc import AsyncIterator , Callable
9- from functools import partial
109from typing import Any
1110
1211import pytest
@@ -126,11 +125,8 @@ async def with_api_server_celery_worker(
126125
127126 app_server = FastAPIAppServer (app = create_app (app_settings ))
128127
129- def _on_worker_init_wrapper (sender : WorkController , ** _kwargs ):
130- assert app_settings .API_SERVER_CELERY # nosec
131- return partial (on_worker_init , app_server , app_settings .API_SERVER_CELERY )(
132- sender , ** _kwargs
133- )
128+ def _on_worker_init_wrapper (sender : WorkController , ** kwargs ):
129+ return on_worker_init (sender , app_server = app_server , ** kwargs )
134130
135131 worker_init .connect (_on_worker_init_wrapper )
136132 worker_shutdown .connect (on_worker_shutdown )
You can’t perform that action at this time.
0 commit comments