Merge pull request #477 from mjwen/feat_sshtunnel

Allow SSHTunnel in job stores
materialsproject · Dec 5, 2023 · 2e24ce1 · 2e24ce1
2 parents e1f4aab + e8fa70c
commit 2e24ce1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/jobflow/core/store.py b/src/jobflow/core/store.py
@@ -661,6 +661,10 @@ def all_subclasses(cl):
 
         all_stores = {s.__name__: s for s in all_subclasses(maggma.stores.Store)}
 
+        # add ssh tunnel support
+        tunnel = maggma.stores.ssh_tunnel.SSHTunnel
+        all_stores[tunnel.__name__] = tunnel
+
         docs_store_info = spec["docs_store"]
         docs_store = _construct_store(docs_store_info, all_stores)