patch to the input iterator

[eacharles]

Problem & Solution Description (including issue #)

Code Quality

• [ ] My code follows the code style of this project
• [ ] I have written unit tests or justified all instances of #pragma: no cover; in the case of a bugfix, a new test that breaks as a result of the bug has been added
• [ ] My code contains relevant comments and necessary documentation for future maintainers; the change is reflected in applicable demos/tutorials (with output cleared!) and added/updated docstrings use the NumPy docstring format
• [ ] Any breaking changes, described above, are accompanied by backwards compatibility and deprecation warnings

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Comparison is base (578d872) 96.96% compared to head (9b39938) 96.96%.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #72 +/- ## ======================================= Coverage 96.96% 96.96% ======================================= Files 32 32 Lines 1714 1714 ======================================= Hits 1662 1662 Misses 52 52 ``` | [Files](https://app.codecov.io/gh/LSSTDESC/rail_base/pull/72?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=LSSTDESC) | Coverage Δ | | |---|---|---| | [src/rail/core/stage.py](https://app.codecov.io/gh/LSSTDESC/rail_base/pull/72?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=LSSTDESC#diff-c3JjL3JhaWwvY29yZS9zdGFnZS5weQ==) | `99.15% <ø> (ø)` | |

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.