From c2e001e34c689d8bac7b3314c946492196647abf Mon Sep 17 00:00:00 2001
From: Tim Repke <repke@mcc-berlin.net>
Date: Wed, 26 Jun 2024 18:51:39 +0200
Subject: [PATCH] fix mypy
---
server/pipelines/actor.py | 2 +-
server/pipelines/tasks/imports.py | 2 +-
server/pipelines/tasks/sleepy.py | 6 +++---
3 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/server/pipelines/actor.py b/server/pipelines/actor.py
index 0825eeb..f1e2979 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 94cde93..b2e394f 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 1b34981..9fccce1 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!')
--
GitLab