patch

materialsproject · Dec 1, 2024 · 3484245 · 3484245
1 parent 0d7bb46
commit 3484245
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/maggma/stores/mongolike.py b/src/maggma/stores/mongolike.py
@@ -446,6 +446,7 @@ def __init__(
         collection_name: str,
         database: Optional[str] = None,
         ssh_tunnel: Optional[SSHTunnel] = None,
+        safe_update: bool = False,
         mongoclient_kwargs: Optional[dict] = None,
         default_sort: Optional[dict[str, Union[Sort, int]]] = None,
         **kwargs,
@@ -461,6 +462,7 @@ def __init__(
         self.uri = uri
         self.ssh_tunnel = ssh_tunnel
         self.default_sort = default_sort
+        self.safe_update = safe_update
         self.mongoclient_kwargs = mongoclient_kwargs or {}
 
         # parse the dbname from the uri