Reduce number of __getitem__ calls (#2420)

Quantum-Accelerators · Aug 13, 2024 · c8e17b5 · c8e17b5
1 parent a9a6362
commit c8e17b5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/src/quacc/recipes/espresso/bands.py b/src/quacc/recipes/espresso/bands.py
@@ -315,13 +315,14 @@ def bands_flow(
         force_gamma=force_gamma,
     )
     results = {"bands_pw": bands_results}
+    bands_results_dir = bands_results["dir_name"]
 
     if run_bands_pp:
-        bands_pp_results = bands_pp_job_(prev_outdir=bands_results["dir_name"])
+        bands_pp_results = bands_pp_job_(prev_outdir=bands_results_dir)
         results["bands_pp"] = bands_pp_results
 
     if run_fermi_surface:
-        fermi_results = fermi_surface_job_(prev_outdir=bands_results["dir_name"])
+        fermi_results = fermi_surface_job_(prev_outdir=bands_results_dir)
         results["fermi_surface"] = fermi_results
 
     return results
diff --git a/src/quacc/recipes/espresso/dos.py b/src/quacc/recipes/espresso/dos.py
@@ -206,8 +206,9 @@ def dos_flow(
     )
 
     static_results = static_job_(atoms)
-    non_scf_results = non_scf_job_(atoms, prev_outdir=static_results["dir_name"])
-    dos_results = dos_job_(prev_outdir=static_results["dir_name"])
+    static_results_dir = static_results["dir_name"]
+    non_scf_results = non_scf_job_(atoms, prev_outdir=static_results_dir)
+    dos_results = dos_job_(prev_outdir=static_results_dir)
 
     return {
         "static_job": static_results,
@@ -282,8 +283,9 @@ def projwfc_flow(
     )
 
     static_results = static_job_(atoms)
-    non_scf_results = non_scf_job_(atoms, prev_outdir=static_results["dir_name"])
-    projwfc_results = projwfc_job_(prev_outdir=static_results["dir_name"])
+    static_results_dir = static_results["dir_name"]
+    non_scf_results = non_scf_job_(atoms, prev_outdir=static_results_dir)
+    projwfc_results = projwfc_job_(prev_outdir=static_results_dir)
 
     return {
         "static_job": static_results,