Flaky test: tests/test_repair.py

[abetomo]

Describe the problem you got

The test sometimes fails like this:

============================= test session starts ==============================
platform linux -- Python 3.12.3, pytest-9.0.3, pluggy-1.6.0
rootdir: /home/runner/work/openarm_dataset/openarm_dataset
configfile: pyproject.toml
testpaths: tests
plugins: anyio-4.13.0
collected 170 items / 1 skipped

tests/test_camera.py ....                                                [  2%]
tests/test_dataset_0_1_0.py ..........                                   [  8%]
tests/test_dataset_0_1_0_to_0_2_0.py ..........                          [ 14%]
tests/test_dataset_0_1_0_to_0_3_0.py ...........                         [ 20%]
tests/test_dataset_0_2_0.py ..........                                   [ 26%]
tests/test_dataset_0_2_0_to_0_3_0.py ...........                         [ 32%]
tests/test_dataset_0_3_0.py ........                                     [ 37%]
tests/test_dataset_unversioned.py ..........                             [ 43%]
tests/test_dataset_unversioned_to_0_2_0.py ..........                    [ 49%]
tests/test_dataset_unversioned_to_0_3_0.py ...........                   [ 55%]
tests/test_gr00t.py .                                                    [ 56%]
tests/test_lerobot_v21.py .......                                        [ 60%]
tests/test_merge.py ...............                                      [ 69%]
tests/test_metadata_0_1_0.py ...........                                 [ 75%]
tests/test_metadata_0_2_0.py ............                                [ 82%]
tests/test_metadata_unversioned.py ...........                           [ 89%]
tests/test_repair.py ........F                                           [ 94%]
tests/test_sampler.py .                                                  [ 95%]
tests/test_validate.py ........                                          [100%]

=================================== FAILURES ===================================
_________________________ test_repair_cli_output_flag __________________________

tmp_path = PosixPath('/tmp/pytest-of-runner/pytest-0/test_repair_cli_output_flag0')

    def test_repair_cli_output_flag(tmp_path):
        dataset = _copy_dataset(tmp_path)
        state_path = dataset / STATE_REL
        _inject_nan_inside_list(state_path, frame=5, joint=0)
        output = tmp_path / "repaired"
    
        result = subprocess.run(
            [
                sys.executable,
                "-m",
                "openarm_dataset.repair",
                str(dataset),
                "-o",
                str(output),
            ],
            capture_output=True,
            text=True,
        )
>       assert result.returncode == 0
E       AssertionError: assert -6 == 0
E        +  where -6 = CompletedProcess(args=['/home/runner/work/openarm_dataset/openarm_dataset/.venv/bin/python', '-m', 'openarm_dataset.re...pisodes/0/obs/arms/left/state.parquet: repaired 1 value(s)\n', stderr='terminate called without an active exception\n').returncode

tests/test_repair.py:192: AssertionError
=========================== short test summary info ============================
SKIPPED [1] tests/test_rrd.py:17: module 'rerun' has __version__ '0.22.1', required is: '0.23'
================== 1 failed, 169 passed, 1 skipped in 38.99s ===================

https://github.com/enactic/openarm_dataset/actions/runs/27175955475/job/80224772434#step:7:60