Fix gridfs URI store

materialsproject · May 24, 2022 · 640e184 · 640e184
1 parent be3c881
commit 640e184
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/maggma/stores/gridfs.py b/src/maggma/stores/gridfs.py
@@ -467,6 +467,8 @@ def __init__(
         compression: bool = False,
         ensure_metadata: bool = False,
         searchable_fields: List[str] = None,
+        auth_source: Optional[str] = None,
+        mongoclient_kwargs: Optional[Dict] = None,
         **kwargs,
     ):
         """
@@ -499,6 +501,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"