chore: Fixed mypy linting

unipoll · Dec 8, 2023 · 42e6e4a · 42e6e4a
1 parent 73aa72b
commit 42e6e4a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/unipoll_api/mongo_db.py b/src/unipoll_api/mongo_db.py
@@ -1,10 +1,11 @@
-import motor.motor_asyncio  # type: ignore
+from motor.core import AgnosticClient
+from motor.motor_asyncio import AsyncIOMotorClient
 from unipoll_api import documents as Documents
 from unipoll_api.config import get_settings
 
 settings = get_settings()
 
-client = motor.motor_asyncio.AsyncIOMotorClient(
+client: AgnosticClient = AsyncIOMotorClient(
     host=settings.mongodb_url,
     uuidRepresentation="standard",
 )