Skip to content

Commit b29cc33

Browse files
committed
Update run_abacus_calculation to return final STRU if relax calculation is not converged
1 parent 5e6b8aa commit b29cc33

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

src/abacusagent/modules/tool_wrapper.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,9 @@ def run_abacus_calculation(
167167
if relax_outputs['result']['normal_end'] is False:
168168
raise ValueError('Relaxation calculation failed')
169169
elif relax_outputs['result']['relax_converge'] is False:
170-
raise ValueError(f'Relaxation calculation did not converge in {max_steps} steps')
170+
return {"msg":f'Relaxation calculation did not converge in {max_steps} steps',
171+
"final_stru": Path(relax_outputs['results']['new_abacus_inputs_dir']) / "STRU",
172+
**relax_outputs["results"]}
171173
else:
172174
print("Relax calculation completed successfully.")
173175
abacus_inputs_dir = relax_outputs['new_abacus_inputs_dir']

0 commit comments

Comments
 (0)