typing fixes, ruff, update deps (#73)

* typing fixes, ruff, update deps * move ruff to dev group
andriykohut · May 28, 2023 · cef1792 · cef1792
1 parent d411608
commit cef1792
Show file tree

Hide file tree

Showing 5 changed files with 135 additions and 186 deletions.
diff --git a/.flake8 b/.flake8
diff --git a/Makefile b/Makefile
@@ -18,7 +18,7 @@ pretty:
 lint:
 	poetry run black --check .
 	poetry run isort --check-only .
-	poetry run flake8 .
+	poetry run ruff check .
 	poetry run mypy .
 
 test:

diff --git a/esdb/client.py b/esdb/client.py
@@ -8,9 +8,10 @@
 import logging
 import random
 import urllib.parse
+from collections.abc import AsyncIterator
 from dataclasses import dataclass
 from functools import cmp_to_key
-from typing import AsyncContextManager, Optional
+from typing import Optional
 
 import grpc
 
@@ -230,8 +231,8 @@ def _create_channel(self, endpoint: Member.Endpoint) -> grpc.aio.Channel:  # typ
             f"{endpoint.address}:{endpoint.port}", credentials, self.options
         )
 
-    @contextlib.asynccontextmanager  # type: ignore
-    async def connect(self) -> AsyncContextManager[Connection]:  # type: ignore
+    @contextlib.asynccontextmanager
+    async def connect(self) -> AsyncIterator[Connection]:
         if self.config.dns_discover:
             endpoint = await self.discover_endpoint()
         else: