Update the changes to resolve the linter issues.

dymmond · Apr 29, 2024 · be4be19 · be4be19
1 parent 81d4213
commit be4be19
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/mongoz/core/db/documents/document.py b/mongoz/core/db/documents/document.py
@@ -21,7 +21,7 @@ class Document(DocumentRow):
     Representation of an Mongoz Document.
     """
 
-    async def create(self: "Document", collection: AsyncIOMotorCollection = None) -> "Document":
+    async def create(self: "Document", collection: AsyncIOMotorCollection = None) -> "Document":  # type: ignore
         """
         Inserts a document.
         """

diff --git a/mongoz/core/db/querysets/core/manager.py b/mongoz/core/db/querysets/core/manager.py
@@ -65,7 +65,7 @@ def __init__(
     def __get__(self, instance: Any, owner: Any) -> "Manager":
         return self.__class__(model_class=owner)
 
-    def using(self, database_name: str):
+    def using(self, database_name: str) -> "Manager":
         """
             **Type** Public
 
@@ -87,7 +87,7 @@ class registry using the database_name that provided in \
                     self._collection.name
                 - return the self instance.
         """
-        database = self.model_class.meta.registry.get_database(database_name)
+        database = self.model_class.meta.registry.get_database(database_name) # type: ignore
         self._collection = database.get_collection(
             self._collection.name)._collection
         return self

diff --git a/tests/models/manager/test_using.py b/tests/models/manager/test_using.py
@@ -6,7 +6,7 @@
 
 import mongoz
 from mongoz import Document, ObjectId
-from mongoz.exceptions import DocumentNotFound, MultipleDocumentsReturned
+from mongoz.exceptions import DocumentNotFound
 
 pytestmark = pytest.mark.anyio
 pydantic_version = pydantic.__version__[:3]