pin pymongo, revert other changes

imaginate-ai · Oct 26, 2024 · 48a9f4e · 48a9f4e
1 parent fe08ef3
commit 48a9f4e
Show file tree

Hide file tree

Showing 3 changed files with 57 additions and 67 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -11,7 +11,7 @@ repository = "https://github.com/imaginate-ai/imaginate-api"
 [tool.poetry.dependencies]
 python = "^3.12"
 Flask = "^2.0"
-pymongo = "^4.0"
+pymongo = "<4.9.0"
 python-dotenv = "^0.21"
 image-handler-client = {git = "https://github.com/imaginate-ai/image-handler-client.git", rev = "v1.0.0"}
 requests = "^2.32.3"

diff --git a/tests/unit/test_app.py b/tests/unit/test_app.py
@@ -27,7 +27,6 @@
 # Mocking
 from unittest.mock import patch, MagicMock
 import mongomock  # NOTE: mongomock doesn't work well with sorting
-mongomock.gridfs.enable_gridfs_integration()
 
 # A struct class to convert a dictionary to a object
 # This is useful since MongoDB uses attributes and does not support dictionary keys