Skip to content

Commit 3af18ad

Browse files
committed
forward kwargs to ElectronicStructureRester
1 parent 5a5a190 commit 3af18ad

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

mp_api/client/routes/materials/electronic_structure.py

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -263,7 +263,14 @@ def get_bandstructure_from_material_id(
263263
bandstructure (Union[BandStructure, BandStructureSymmLine]): BandStructure or BandStructureSymmLine object
264264
"""
265265
es_rester = ElectronicStructureRester(
266-
endpoint=self.base_endpoint, api_key=self.api_key
266+
api_key=self.api_key,
267+
endpoint=self.base_endpoint,
268+
include_user_agent=self.include_user_agent,
269+
session=self.session,
270+
monty_decode=self.monty_decode,
271+
use_document_model=self.use_document_model,
272+
headers=self.headers,
273+
mute_progress_bars=self.mute_progress_bars,
267274
)
268275

269276
if line_mode:
@@ -445,7 +452,14 @@ def get_dos_from_material_id(self, material_id: str):
445452
dos (CompleteDos): CompleteDos object
446453
"""
447454
es_rester = ElectronicStructureRester(
448-
endpoint=self.base_endpoint, api_key=self.api_key
455+
api_key=self.api_key,
456+
endpoint=self.base_endpoint,
457+
include_user_agent=self.include_user_agent,
458+
session=self.session,
459+
monty_decode=self.monty_decode,
460+
use_document_model=self.use_document_model,
461+
headers=self.headers,
462+
mute_progress_bars=self.mute_progress_bars,
449463
)
450464

451465
dos_doc = es_rester.search(material_ids=material_id, fields=["dos"])

0 commit comments

Comments
 (0)