diff --git a/src/taskiq_deduplication/middleware.py b/src/taskiq_deduplication/middleware.py index 3e8a1e8..29c8cd5 100644 --- a/src/taskiq_deduplication/middleware.py +++ b/src/taskiq_deduplication/middleware.py @@ -2,8 +2,7 @@ import asyncio import hashlib import json import logging -from collections.abc import Awaitable -from typing import Any, cast +from typing import Any from pydantic import RedisDsn from redis.asyncio import Redis @@ -105,7 +104,7 @@ class RedisDeduplicationMiddleware(TaskiqMiddleware): for attempt in range(self.startup_retries): client = Redis.from_url(str(self.redis_url)) try: - await cast(Awaitable[bool], client.ping()) + await client.ping() self._redis = client self._release_script = self._redis.register_script(RELEASE_LUA_SCRIPT) self._refresh_script = self._redis.register_script(REFRESH_LUA_SCRIPT) diff --git a/tests/conftest.py b/tests/conftest.py index e5bbb6b..a78cae5 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,5 +1,3 @@ -from typing import Awaitable, cast - import pytest import fakeredis.aioredis from redis.asyncio import Redis @@ -22,7 +20,7 @@ async def fake_redis(): async def real_redis(): client = Redis.from_url("redis://localhost:6379/15") try: - await cast(Awaitable[bool], client.ping()) + await client.ping() except Exception: await client.aclose() pytest.skip("Redis not available at localhost:6379")