diff --git a/src/maggma/stores/mongolike.py b/src/maggma/stores/mongolike.py index 40e9a1856..8c257cdd6 100644 --- 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