Fix passing session and headers in base rester (#717)

* Fix passing session and headers in base rester * Linting * More linting
materialsproject · Dec 8, 2022 · 8f21dbb · 8f21dbb
1 parent ac4bb36
commit 8f21dbb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mp_api/client/core/client.py b/mp_api/client/core/client.py
@@ -845,7 +845,8 @@ def get_data_by_id(
                 from mp_api.client.routes.materials import MaterialsRester
 
                 with MaterialsRester(
-                    api_key=self.api_key, endpoint=self.base_endpoint, use_document_model=False, monty_decode=False
+                    api_key=self.api_key, endpoint=self.base_endpoint, use_document_model=False, monty_decode=False,
+                    session=self.session, headers=self.headers
                 ) as mpr:
                     docs = mpr.search(task_ids=[document_id], fields=["material_id"])