handle store error during finalize

materialsproject · May 29, 2024 · 819d187 · 819d187
1 parent f88951d
commit 819d187
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/maggma/core/builder.py b/src/maggma/core/builder.py
@@ -9,7 +9,7 @@
 
 from monty.json import MontyDecoder, MSONable
 
-from maggma.core.store import Store
+from maggma.core.store import Store, StoreError
 from maggma.utils import TqdmLoggingHandler, grouper, tqdm
 
 
@@ -114,7 +114,7 @@ def finalize(self):
         for store in self.sources + self.targets:
             try:
                 store.close()
-            except AttributeError:
+            except (AttributeError, StoreError):
                 continue
 
     def run(self, log_level=logging.DEBUG):