Merge pull request #516 from materialsproject/mongo_patch

Patch mongo store connect methods
materialsproject · Dec 1, 2021 · 603c930 · 603c930
2 parents 634afe3 + b9cf95c
commit 603c930
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/maggma/stores/mongolike.py b/src/maggma/stores/mongolike.py
@@ -149,7 +149,7 @@ def connect(self, force_reset: bool = False):
         """
         Connect to the source data
         """
-        if not self._collection or force_reset:
+        if self._collection is None or force_reset:
             if self.ssh_tunnel is None:
                 conn = MongoClient(self.host, self.port)
             else:
@@ -549,7 +549,7 @@ def connect(self, force_reset: bool = False):
         Connect to the source data
         """
 
-        if not self._collection or force_reset:
+        if self._collection is None or force_reset:
             self._collection = mongomock.MongoClient().db[self.name]
 
     @property