using base attribute to get number_of_bands

aiidateam · Nov 20, 2024 · 2a94358 · 2a94358
1 parent d4057cc
commit 2a94358
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/aiida_quantumespresso/workflows/pdos.py b/src/aiida_quantumespresso/workflows/pdos.py
@@ -434,7 +434,7 @@ def run_nscf(self):
 
             if 'nbands_factor' in self.inputs:
                 factor = self.inputs.nbands_factor.value
-                parameters = self.ctx.workchain_scf.outputs.output_parameters.get_dict()
+                parameters = self.ctx.workchain_scf.outputs.output_parameters.base.attributes.get('number_of_bands')
                 nbands = int(parameters['number_of_bands'])
                 nelectron = int(parameters['number_of_electrons'])
                 nbnd = max(int(0.5 * nelectron * factor), int(0.5 * nelectron) + 4, nbands)