fix in plot_fits for some edge cases

trichter · Jan 17, 2023 · ed58f48 · ed58f48
1 parent ce0db15
commit ed58f48
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/qopen/imaging.py b/qopen/imaging.py
@@ -326,8 +326,7 @@ def get_Emod(G, t):
         G_ = smooth_func(lambda t_: G_func(r, t_, v0, g0),
                          t, smooth, window=smooth_window)
         Emod = get_Emod(G_, t)
-        index = np.argwhere(Emod < 1e-30)[-1]
-        Emod[index] = 1e-30
+        Emod[Emod < 1e-30] = 1e-30
         plot(t, Emod, color=c2l)
 
         plot(tcoda[i], Ecoda[i], color=c1)