switch from mongomock persistence branch to mongomock-persistence pac…

…kage
materialsproject · May 24, 2024 · 8b27956 · 8b27956
1 parent 605a009
commit 8b27956
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/fireworks/fw_config.py b/fireworks/fw_config.py
@@ -8,7 +8,7 @@
 from monty.design_patterns import singleton
 from monty.serialization import dumpfn, loadfn
 import pymongo
-import mongomock
+import mongomock_persistence
 import mongomock.gridfs
 
 __author__ = "Anubhav Jain"
@@ -169,7 +169,7 @@ def override_user_settings() -> None:
     if globals()['MONGOMOCK_SERVERSTORE_FILE']:
         if not os.environ.get('MONGOMOCK_SERVERSTORE_FILE'):
             os.environ['MONGOMOCK_SERVERSTORE_FILE'] = globals()['MONGOMOCK_SERVERSTORE_FILE']
-        globals()['MongoClient'] = getattr(mongomock, 'MongoClient')
+        globals()['MongoClient'] = getattr(mongomock_persistence, 'MongoClient')
         if globals()['GRIDFS_FALLBACK_COLLECTION']:
             mongomock.gridfs.enable_gridfs_integration()
 

diff --git a/requirements.txt b/requirements.txt
@@ -8,4 +8,4 @@ flask==2.2.5
 flask-paginate==0.5.5
 gunicorn==20.0.4
 tqdm==4.41.0
-mongomock @ git+https://github.com/ikondov/mongomock.git@persistence
+mongomock-persistence==0.0.3
diff --git a/setup.py b/setup.py
@@ -39,7 +39,7 @@
             "tqdm>=4.8.4",
             "importlib-metadata>=4.8.2; python_version<'3.8'",
             "typing-extensions; python_version<'3.8'",
-            "mongomock @ git+https://github.com/ikondov/mongomock.git@persistence",
+            "mongomock-persistence>=0.0.3",
         ],
         extras_require={
             "rtransfer": ["paramiko>=2.4.2"],