Fixed mypy.

Author: s3rius

taskiq_redis/list_schedule_source.py

@@ -130,7 +130,7 @@ async def _get_previous_time_schedules(self) -> list[bytes]:
                 if key_time and key_time <= minute_before:
                     time_keys.append(key.decode())
             for key in time_keys:
-                schedules.extend(await redis.lrange(key, 0, -1))  # type: ignore
+                schedules.extend(await redis.lrange(key, 0, -1))
 
         return schedules
 
@@ -146,10 +146,10 @@ async def delete_schedule(self, schedule_id: str) -> None:
                 )
                 # We need to remove the schedule from the cron or time list.
                 if schedule.cron is not None:
-                    await redis.lrem(self._get_cron_key(), 0, schedule_id)  # type: ignore
+                    await redis.lrem(self._get_cron_key(), 0, schedule_id)
                 elif schedule.time is not None:
                     time_key = self._get_time_key(schedule.time)
-                    await redis.lrem(time_key, 0, schedule_id)  # type: ignore
+                    await redis.lrem(time_key, 0, schedule_id)
 
     async def add_schedule(self, schedule: "ScheduledTask") -> None:
         """Add a schedule to the source."""
@@ -163,9 +163,9 @@ async def add_schedule(self, schedule: "ScheduledTask") -> None:
             # This is an optimization, so we can get all the schedules
             # for the current time much faster.
             if schedule.cron is not None:
-                await redis.rpush(self._get_cron_key(), schedule.schedule_id)  # type: ignore
+                await redis.rpush(self._get_cron_key(), schedule.schedule_id)
             elif schedule.time is not None:
-                await redis.rpush(  # type: ignore
+                await redis.rpush(
                     self._get_time_key(schedule.time),
                     schedule.schedule_id,
                 )
@@ -195,11 +195,11 @@ async def get_schedules(self) -> List["ScheduledTask"]:
             self._is_first_run = False
         async with Redis(connection_pool=self._connection_pool) as redis:
             buffer = []
-            crons = await redis.lrange(self._get_cron_key(), 0, -1)  # type: ignore
+            crons = await redis.lrange(self._get_cron_key(), 0, -1)
             logger.debug("Got %d cron schedules", len(crons))
             if crons:
                 buffer.extend(crons)
-            timed.extend(await redis.lrange(self._get_time_key(current_time), 0, -1))  # type: ignore
+            timed.extend(await redis.lrange(self._get_time_key(current_time), 0, -1))
             logger.debug("Got %d timed schedules", len(timed))
             if timed:
                 buffer.extend(timed)

taskiq_redis/redis_backend.py

@@ -131,7 +131,7 @@ async def set_result(
             redis_set_params["px"] = self.result_px_time
 
         async with Redis(connection_pool=self.redis_pool) as redis:
-            await redis.set(**redis_set_params)  # type: ignore
+            await redis.set(**redis_set_params)
 
     async def is_result_ready(self, task_id: str) -> bool:
         """
@@ -205,7 +205,7 @@ async def set_progress(
             redis_set_params["px"] = self.result_px_time
 
         async with Redis(connection_pool=self.redis_pool) as redis:
-            await redis.set(**redis_set_params)  # type: ignore
+            await redis.set(**redis_set_params)
 
     async def get_progress(
         self,

taskiq_redis/redis_broker.py

@@ -122,7 +122,7 @@ async def kick(self, message: BrokerMessage) -> None:
         """
         queue_name = message.labels.get("queue_name") or self.queue_name
         async with Redis(connection_pool=self.connection_pool) as redis_conn:
-            await redis_conn.lpush(queue_name, message.message)  # type: ignore
+            await redis_conn.lpush(queue_name, message.message)
 
     async def listen(self) -> AsyncGenerator[bytes, None]:
         """
@@ -137,7 +137,7 @@ async def listen(self) -> AsyncGenerator[bytes, None]:
         while True:
             try:
                 async with Redis(connection_pool=self.connection_pool) as redis_conn:
-                    yield (await redis_conn.brpop(self.queue_name))[  # type: ignore
+                    yield (await redis_conn.brpop(self.queue_name))[
                         redis_brpop_data_position
                     ]
             except ConnectionError as exc:
@@ -280,7 +280,7 @@ async def listen(self) -> AsyncGenerator[AckableMessage, None]:
                     self.consumer_name,
                     {
                         self.queue_name: ">",
-                        **self.additional_streams,  # type: ignore
+                        **self.additional_streams,
                     },
                     block=self.block,
                     noack=False,