diff --git a/src/mongo/mongo.py b/src/mongo/mongo.py index 294fb3c..dbeec7e 100644 --- a/src/mongo/mongo.py +++ b/src/mongo/mongo.py @@ -27,7 +27,7 @@ import logging from typing import Any, List, Optional from pymongo import ReturnDocument -from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorCollection +from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorCollection, AsyncIOMotorDatabase log = logging.getLogger(__name__) @@ -48,7 +48,7 @@ class Mongo: return self._db[name] @property - def database(self): + def database(self) -> AsyncIOMotorDatabase: """raw motor database handle""" return self._db