Store.connect: only reset ssh if force_reset True

materialsproject · Nov 4, 2023 · 9ed8cb8 · 9ed8cb8
1 parent 80585fb
commit 9ed8cb8
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 14 deletions.
diff --git a/src/maggma/stores/gridfs.py b/src/maggma/stores/gridfs.py
@@ -132,14 +132,14 @@ def connect(self, force_reset: bool = False):
             force_reset: whether to reset the connection or not when the Store is
                 already connected.
         """
-        if self.ssh_tunnel is None:
-            host = self.host
-            port = self.port
-        else:
-            self.ssh_tunnel.start()
-            host, port = self.ssh_tunnel.local_address
-
         if not self._coll or force_reset:
+            if self.ssh_tunnel is None:
+                host = self.host
+                port = self.port
+            else:
+                self.ssh_tunnel.start()
+                host, port = self.ssh_tunnel.local_address
+
             conn: MongoClient = (
                 MongoClient(
                     host=host,

diff --git a/src/maggma/stores/mongolike.py b/src/maggma/stores/mongolike.py
@@ -179,14 +179,14 @@ def connect(self, force_reset: bool = False):
             force_reset: whether to reset the connection or not when the Store is
                 already connected.
         """
-        if self.ssh_tunnel is None:
-            host = self.host
-            port = self.port
-        else:
-            self.ssh_tunnel.start()
-            host, port = self.ssh_tunnel.local_address
-
         if self._coll is None or force_reset:
+            if self.ssh_tunnel is None:
+                host = self.host
+                port = self.port
+            else:
+                self.ssh_tunnel.start()
+                host, port = self.ssh_tunnel.local_address
+
             conn: MongoClient = (
                 MongoClient(
                     host=host,