Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions beanie/odm/documents.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,12 +23,10 @@
from bson import DBRef, ObjectId
from lazy_model import LazyModel
from pydantic import (
ConfigDict,
Field,
PrivateAttr,
ValidationError,
)
from pydantic.class_validators import root_validator
from pydantic.main import BaseModel
from pymongo import InsertOne
from pymongo.asynchronous.client_session import AsyncClientSession
Expand Down Expand Up @@ -109,7 +107,9 @@
from beanie.odm.utils.typing import extract_id_class

if IS_PYDANTIC_V2:
from pydantic import model_validator
from pydantic import ConfigDict, model_validator
else:
from pydantic.class_validators import root_validator

if TYPE_CHECKING:
from beanie.odm.views import View
Expand Down
1 change: 1 addition & 0 deletions beanie/odm/queries/find.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,6 +153,7 @@ async def count(self) -> int:
kwargs["limit"] = self.limit_number
if self.skip_number:
kwargs["skip"] = self.skip_number
kwargs.update(self.pymongo_kwargs)
return (
await self.document_model.get_pymongo_collection().count_documents(
self.get_filter_query(), session=self.session, **kwargs
Expand Down
Loading