fix mypy

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]

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!')

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!')