make mongo client configurable (#83)

* make mongo client configurable * name properly Co-authored-by: Rüdiger Busche <[email protected]> * Update incense/experiment_loader.py Co-authored-by: Rüdiger Busche <[email protected]>
JarnoRFB · Dec 30, 2022 · 21f7cbb · 21f7cbb
1 parent 7c9f229
commit 21f7cbb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/incense/experiment_loader.py b/incense/experiment_loader.py
@@ -17,8 +17,8 @@
 class ExperimentLoader:
     """Loads artifacts related to experiments."""
 
-    def __init__(self, mongo_uri=None, db_name="sacred", unpickle: bool = True):
-        client: MongoClient = MongoClient(mongo_uri)
+    def __init__(self, mongo_uri=None, db_name="sacred", unpickle: bool = True, **mongo_client_kwargs):
+        client: MongoClient = MongoClient(mongo_uri, **mongo_client_kwargs)
         self._database = client[db_name]
         self._runs = self._database.runs
         self._grid_filesystem = gridfs.GridFS(self._database)