diff --git a/pySDC/projects/PinTSimE/estimation_check.py b/pySDC/projects/PinTSimE/estimation_check.py index 843dbba655..de05b8d6b3 100644 --- a/pySDC/projects/PinTSimE/estimation_check.py +++ b/pySDC/projects/PinTSimE/estimation_check.py @@ -243,9 +243,8 @@ def plotStateFunctionAroundEvent(u_num, prob_cls_name, M_fix): # pragma: no cov if use_SE: t_switches = [u_num[dt][M_fix][use_SE][use_A]['t_switches'] for dt in dt_list] for t_switch_item in t_switches: - for m in range(1, len(t_switch_item)): - mask = np.append([True], np.abs(t_switch_item[1:] - t_switch_item[:-1]) > 1e-10) - t_switch_item = t_switch_item[mask] + mask = np.append([True], np.abs(t_switch_item[1:] - t_switch_item[:-1]) > 1e-10) + t_switch_item = t_switch_item[mask] t_switch = [t_event[i] for t_event in t_switches] ax[0, ind].plot(