Skip to content

Commit ad768be

Browse files
fix: tests
1 parent 870d96c commit ad768be

File tree

1 file changed

+2
-6
lines changed
  • services/api-server/tests/unit/api_functions/celery

1 file changed

+2
-6
lines changed

services/api-server/tests/unit/api_functions/celery/conftest.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66

77
import datetime
88
from collections.abc import AsyncIterator, Callable
9-
from functools import partial
109
from typing import Any
1110

1211
import 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)

0 commit comments

Comments
 (0)