diff --git a/src/maggma/stores/gridfs.py b/src/maggma/stores/gridfs.py index db5d574d1..3122a0b1b 100644 --- a/src/maggma/stores/gridfs.py +++ b/src/maggma/stores/gridfs.py @@ -467,6 +467,7 @@ def __init__( compression: bool = False, ensure_metadata: bool = False, searchable_fields: List[str] = None, + mongoclient_kwargs: Optional[Dict] = None, **kwargs, ): """ @@ -499,6 +500,7 @@ def __init__( self.ensure_metadata = ensure_metadata self.searchable_fields = [] if searchable_fields is None else searchable_fields self.kwargs = kwargs + self.mongoclient_kwargs = mongoclient_kwargs or {} if "key" not in kwargs: kwargs["key"] = "_id"