Merge pull request #5 from DS4SD/chore-update-dependencies

chore: Update dependencies

mognet/backend/redis_result_backend.py

@@ -10,16 +10,17 @@
 from typing import Any, AnyStr, Dict, Iterable, List, Optional, Set
 from uuid import UUID
 
-import aioredis
-from aioredis.exceptions import TimeoutError, ConnectionError
+from pydantic.tools import parse_raw_as
+from redis.asyncio import Redis, from_url
+from redis.exceptions import ConnectionError, TimeoutError
+
 from mognet.backend.backend_config import Encoding, ResultBackendConfig
 from mognet.backend.base_result_backend import AppParameters, BaseResultBackend
 from mognet.exceptions.base_exceptions import NotConnected
 from mognet.exceptions.result_exceptions import ResultValueLost
 from mognet.model.result import Result, ResultValueHolder
 from mognet.model.result_state import READY_STATES, ResultState
 from mognet.tools.urls import censor_credentials
-from pydantic.tools import parse_raw_as
 
 _log = logging.getLogger(__name__)
 
@@ -83,7 +84,7 @@ def __init__(self, config: ResultBackendConfig, app: AppParameters) -> None:
         self._retry_lock = asyncio.Lock()
 
     @property
-    def _redis(self) -> aioredis.Redis:
+    def _redis(self) -> Redis:
         if self.__redis is None:
             raise NotConnected
 
@@ -402,7 +403,7 @@ async def _close_waiters(self):
 
     async def _create_redis(self):
         _log.debug("Creating Redis connection")
-        redis: aioredis.Redis = await aioredis.from_url(
+        redis: Redis = await from_url(
             self._url,
             max_connections=self.config.redis.max_connections,
         )

mognet/state/redis_state_backend.py

@@ -1,10 +1,12 @@
-import logging
 import json
-from typing import Any, Optional, TYPE_CHECKING, TypeVar
-import aioredis
+import logging
+from typing import TYPE_CHECKING, Any, Optional, TypeVar
+
+from redis.asyncio import Redis, from_url
+
 from mognet.exceptions.base_exceptions import NotConnected
-from mognet.state.state_backend_config import StateBackendConfig
 from mognet.state.base_state_backend import BaseStateBackend
+from mognet.state.state_backend_config import StateBackendConfig
 from mognet.tools.urls import censor_credentials
 
 if TYPE_CHECKING:
@@ -25,7 +27,7 @@ def __init__(self, config: StateBackendConfig, app: "App") -> None:
         self.app = app
 
     @property
-    def _redis(self) -> aioredis.Redis:
+    def _redis(self) -> Redis:
         if self.__redis is None:
             raise NotConnected
 
@@ -110,7 +112,7 @@ async def __aexit__(self, *args, **kwargs):
         await self.close()
 
     async def connect(self):
-        redis: aioredis.Redis = aioredis.from_url(
+        redis: Redis = from_url(
             self.config.redis.url,
             max_connections=self.config.redis.max_connections,
         )