Function _count_steps_in_range

modules/processing_correction.py:196–202

Count scheduler timesteps that fall within (low, high) exclusive.

(pipe, low, high)

def _count_steps_in_range(pipe, low, high):
    """Count scheduler timesteps that fall within (low, high) exclusive."""
    timesteps = getattr(getattr(pipe, 'scheduler', None), 'timesteps', None)
    if timesteps is None:
        return 1
    count = sum(1 for t in timesteps.tolist() if low < t < high)
    return max(count, 1)


def _count_steps_below(pipe, threshold):