Add return type

safe-global · Jan 16, 2025 · 133a83b · 133a83b
1 parent bbb9171
commit 133a83b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/app/datasources/db/models.py b/app/datasources/db/models.py
@@ -1,5 +1,5 @@
 from datetime import datetime, timezone
-from typing import AsyncIterator, cast
+from typing import AsyncGenerator, AsyncIterator, cast
 
 from sqlmodel import (
     JSON,
@@ -302,7 +302,9 @@ async def get_contracts_without_abi(
             yield contract
 
     @classmethod
-    async def get_proxy_contracts(cls, session: AsyncSession):
+    async def get_proxy_contracts(
+        cls, session: AsyncSession
+    ) -> AsyncGenerator["Contract"]:
         """
         Return all the contracts with implementation address, so proxy contracts.