Skip registering already registered stores (#138)

exalearn · May 29, 2024 · f0cd487 · f0cd487
1 parent f5fcf49
commit f0cd487
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/colmena/proxy.py b/colmena/proxy.py
@@ -36,7 +36,11 @@ def get_store(
     store = proxystore.store.get_store(name)
     if store is None and config is not None:
         store = Store.from_config(config)
-        proxystore.store.register_store(store)
+        # If the config has `register=True`, then the above line will
+        # register the store for us. So we set `exist_ok=True` so the next
+        # line does not raise an error. This only applies for
+        # ProxyStore v0.6.5 and later.
+        proxystore.store.register_store(store, exist_ok=True)
     return store