Skip to content

Commit aa6c313

Browse files
committed
Fix rare case where list of step energies and max_trunc_error_list is empty
1 parent dbdad20 commit aa6c313

File tree

1 file changed

+11
-2
lines changed

1 file changed

+11
-2
lines changed

varipeps/optimization/optimizer.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -599,8 +599,17 @@ def random_noise(a):
599599
enforce_elementwise_convergence=varipeps_config.ad_use_custom_vjp,
600600
)
601601
varipeps_global_state.ctmrg_projector_method = None
602-
max_trunc_error_list[random_noise_retries][-1] = max_trunc_error
603-
step_energies[random_noise_retries][-1] = working_value
602+
603+
try:
604+
max_trunc_error_list[random_noise_retries][-1] = max_trunc_error
605+
except IndexError:
606+
max_trunc_error_list[random_noise_retries].append(max_trunc_error)
607+
608+
try:
609+
step_energies[random_noise_retries][-1] = working_value
610+
except IndexError:
611+
step_energies[random_noise_retries].append(working_value)
612+
604613
break
605614

606615
if (

0 commit comments

Comments
 (0)