Fix BdG call before assignment in submission

JuDFTteam · May 29, 2024 · b9ac704 · b9ac704
1 parent 093c343
commit b9ac704
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/aiida_kkr/workflows/kkr_STM.py b/aiida_kkr/workflows/kkr_STM.py
@@ -485,11 +485,13 @@ def STM_lmdos_run(self):
         y = self.inputs.tip_position['ny']
 
         calc = self.submit(builder)
+
         message = f"""INFO: running DOS step for an STM measurement (pk: {calc.pk}) at position
                           (ilayer: {self.inputs.tip_position['ilayer']}, da: {x}, db: {y} )"""
 
-        if self.inputs.BdG.params_kkr_overwrite:
-            message = f"""INFO: runnig DOS step (pk: {calc.pk}) BdG is present"""
+        if 'BdG' in self.inputs:
+            if self.inputs.BdG.params_kkr_overwrite:
+                message = f"""INFO: runnig DOS step (pk: {calc.pk}) BdG is present"""
 
         print(message)
         self.report(message)