Xiyou Zhou addressed a critical reliability issue in the apple/axlearn repository by fixing an infinite loop in the language-modeling input processor, which previously risked causing training jobs to hang indefinitely. Using Python and leveraging skills in data processing and machine learning, Xiyou identified the root cause related to infinite dataset handling and implemented a targeted solution to prevent repeated data cycling. To ensure long-term robustness, Xiyou also developed a regression test using unit testing practices, enhancing the test suite and continuous integration coverage. This work improved the stability of long-running training workflows and strengthened dataset validation within the project.