Removed loop

Parallel-in-Time · Apr 18, 2024 · 2affa75 · 2affa75
1 parent e8bf17e
commit 2affa75
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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(