Correct bug in step function

NNPDF · Jan 29, 2025 · f916a65 · f916a65
1 parent 2e517f3
commit f916a65
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/validphys2/src/validphys/theorycovariance/higher_twist_functions.py b/validphys2/src/validphys/theorycovariance/higher_twist_functions.py
@@ -74,7 +74,9 @@ def step_function(a: npt.ArrayLike, y_shift: npt.ArrayLike, bin_edges: npt.Array
     for shift_pos, shift in enumerate(y_shift):
         bin_low = bin_edges[shift_pos]
         bin_high = bin_edges[shift_pos + 1]
-        condition = np.multiply(a >= bin_low, a < bin_high)
+        condition = np.multiply(
+            a >= bin_low, a < bin_high if shift_pos != len(y_shift) - 1 else a <= bin_high
+        )
         res.append([shift for cond in condition if cond])
     res = np.concatenate(res)
     return res