Fix type errors

jefersondaniel · Apr 25, 2024 · 3123ed8 · 3123ed8
1 parent bbc6efc
commit 3123ed8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pydantic_mongo/abstract_repository.py b/pydantic_mongo/abstract_repository.py
@@ -121,7 +121,7 @@ def save(self, model: T) -> Union[InsertOneResult, UpdateResult]:
             )
 
         result = self.get_collection().insert_one(document)
-        model.id = result.inserted_id
+        model_with_id.id = result.inserted_id
         return result
 
     def save_many(self, models: Iterable[T]):
@@ -143,7 +143,7 @@ def save_many(self, models: Iterable[T]):
             )
 
             for idx, inserted_id in enumerate(result.inserted_ids):
-                models_to_insert[idx].id = inserted_id
+                cast(ModelWithId, models_to_insert[idx]).id = inserted_id
 
         if len(models_to_update) == 0:
             return

diff --git a/requirements_test.txt b/requirements_test.txt
@@ -7,7 +7,7 @@ pytest==8.1.1
 pytest-cov==4.1.0
 pytest-mock==3.12.0
 mongomock==4.1.2
-pydantic>=2.0.2
+pydantic==2.7.1
 pymongo==4.6.3
 mypy==1.10.0
 mypy-extensions==1.0.0