diff --git a/server/pipelines/actor.py b/server/pipelines/actor.py index 0825eeb5e85a58014ab8afb37a46a70b364c2902..f1e2979340ba5e1d2ded5c3e826c019640e58c5d 100644 --- a/server/pipelines/actor.py +++ b/server/pipelines/actor.py @@ -90,7 +90,7 @@ class NacsosActor(Actor[P, R]): actor_name: str = 'anonymous_actor' task_id: str | None = None message_id: str | None = None - message: Message[R] = CurrentMessage.get_current_message() # type: ignore[type-arg] + message: Message[R] = CurrentMessage.get_current_message() # type: ignore[type-arg,assignment] if message: message_id = message.message_id actor_name = message.options.get('nacsos_actor_name') # type: ignore[assignment] diff --git a/server/pipelines/tasks/imports.py b/server/pipelines/tasks/imports.py index 94cde93310736d8920d68440a370ace7db997ab7..b2e394f299a308c9f8ebd496a9032c48f46fba3e 100644 --- a/server/pipelines/tasks/imports.py +++ b/server/pipelines/tasks/imports.py @@ -23,7 +23,7 @@ def prefix_sources(sources: list[Path]): return [settings.PIPES.user_data_dir / path for path in sources] -@dramatiq.actor(actor_class=NacsosActor, max_retries=0) +@dramatiq.actor(actor_class=NacsosActor, max_retries=0) # type: ignore[arg-type] async def import_task(import_id: str | None = None) -> None: async with NacsosActor.exec_context() as (session, logger, target_dir, work_dir, task_id, message_id): logger.info('Preparing import task!') diff --git a/server/pipelines/tasks/sleepy.py b/server/pipelines/tasks/sleepy.py index 1b349811d49309c6abc3f6d28ffb37562adf3d1d..9fccce110cae6079916af38b81b93a9649e4c0dc 100644 --- a/server/pipelines/tasks/sleepy.py +++ b/server/pipelines/tasks/sleepy.py @@ -7,7 +7,7 @@ from dramatiq.middleware import CurrentMessage from ..actor import NacsosActor -@dramatiq.actor(actor_class=NacsosActor) +@dramatiq.actor(actor_class=NacsosActor) # type: ignore[arg-type] async def tracked_sleep_task(sleep_time: int = 10) -> None: message = CurrentMessage.get_current_message() print('message') @@ -21,12 +21,12 @@ async def tracked_sleep_task(sleep_time: int = 10) -> None: logger.info('Done, yo!') -@dramatiq.actor(queue_name='nacsos-pipes') +@dramatiq.actor(queue_name='nacsos-pipes') # type: ignore[arg-type] async def sleep_task(sleep_time: int = 10) -> None: cm = CurrentMessage.get_current_message() logger = logging.getLogger('sleepy') logger.info('Preparing sleep task!') logger.info(f'{cm}') - logger.info(f'{cm.message_id}') + logger.info(f'{cm.message_id}') # type: ignore[union-attr] await asyncio.sleep(sleep_time) logger.info('Done, yo!')