Merge pull request #1014 from jmmshn/safe_update

Minor fix to URI store
materialsproject · Dec 3, 2024 · dc204f8 · dc204f8
2 parents 668c2e4 + e69ea16
commit dc204f8
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