Revert is_stage_over for performance
the older iterative implementation outperforms the current vectorized numpy implementation. Reverting for the better performance (and more understandable code).
the older iterative implementation outperforms the current vectorized numpy implementation. Reverting for the better performance (and more understandable code).