fix: workflows: asyncio

fsconnectors/asyncio/connector.py

@@ -1,6 +1,6 @@
-from typing import List, Any
 from abc import ABC, abstractmethod
 from contextlib import asynccontextmanager
+from typing import List, Any, AsyncGenerator
 
 from fsconnectors.utils.entry import FSEntry
 
@@ -11,7 +11,7 @@ class AsyncConnector(ABC):
     @classmethod
     @abstractmethod
     @asynccontextmanager
-    async def connect(cls, *args, **kwargs) -> 'AsyncConnector':
+    async def connect(cls, *args, **kwargs) -> AsyncGenerator['AsyncConnector', None]:
         """Connects to file system.
 
         Yields

fsconnectors/asyncio/local.py

@@ -3,8 +3,8 @@
 import aiofiles
 import aioshutil
 import aiofiles.os
-from typing import List, Any
 from contextlib import asynccontextmanager
+from typing import List, Any, AsyncGenerator
 
 from fsconnectors.utils.entry import FSEntry
 from fsconnectors.asyncio.connector import AsyncConnector
@@ -15,7 +15,7 @@ class AsyncLocalConnector(AsyncConnector):
 
     @classmethod
     @asynccontextmanager
-    async def connect(cls) -> 'AsyncLocalConnector':
+    async def connect(cls) -> AsyncGenerator['AsyncLocalConnector', None]:
         """Connects to file system.
 
         Yields

fsconnectors/asyncio/s3.py

@@ -1,8 +1,8 @@
 import yaml
 import aioboto3
 import aiofiles.tempfile
-from typing import List, Any, Literal
 from contextlib import asynccontextmanager
+from typing import List, Any, AsyncGenerator
 
 from fsconnectors.utils.entry import FSEntry
 from fsconnectors.utils.multipart import AsyncMultipartWriter
@@ -19,7 +19,7 @@ async def connect(
         endpoint_url: str,
         aws_access_key_id: str,
         aws_secret_access_key: str
-    ) -> 'AsyncS3Connector':
+    ) -> AsyncGenerator['AsyncS3Connector', None]:
         """Connects to file system.
 
         Parameters