|
| 1 | +name: dataset |
| 2 | + |
| 3 | +on: |
| 4 | + # Trigger the workflow on push or pull request, |
| 5 | + # but only for the main branch |
| 6 | + pull_request: |
| 7 | + branches: |
| 8 | + - main |
| 9 | + paths: |
| 10 | + - "verl/utils/**/*.py" |
| 11 | + - .github/workflows/dataset.yml |
| 12 | + - "!verl/workers/fsdp_workers.py" |
| 13 | + - "!verl/workers/megatron_workers.py" |
| 14 | + - "!recipe/**" |
| 15 | + |
| 16 | +# Cancel jobs on the same ref if a new one is triggered |
| 17 | +concurrency: |
| 18 | + group: ${{ github.workflow }}-${{ github.ref }} |
| 19 | + cancel-in-progress: ${{ github.ref != 'refs/heads/main' }} |
| 20 | + |
| 21 | +# Declare permissions just read content. |
| 22 | +permissions: |
| 23 | + contents: read |
| 24 | + |
| 25 | +jobs: |
| 26 | + ray: |
| 27 | + runs-on: [L20x8] |
| 28 | + timeout-minutes: 10 # Increase this timeout value as needed |
| 29 | + env: |
| 30 | + HTTP_PROXY: ${{ secrets.PROXY_HTTP }} |
| 31 | + HTTPS_PROXY: ${{ secrets.PROXY_HTTPS }} |
| 32 | + NO_PROXY: "localhost,127.0.0.1" |
| 33 | + HF_HUB_ENABLE_HF_TRANSFER: "0" # This is more stable |
| 34 | + container: |
| 35 | + image: whatcanyousee/verl:ngc-th2.6.0-cu124-vllm0.8.2-mcore0.11.0-te2.0 |
| 36 | + options: --gpus all --shm-size=10g |
| 37 | + steps: |
| 38 | + - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2 |
| 39 | + with: |
| 40 | + fetch-depth: 0 |
| 41 | + - name: Install the current repository |
| 42 | + run: | |
| 43 | + pip install -e .[test] |
| 44 | + pip install --upgrade "ray>=2.40.0" |
| 45 | + pip install cupy-cuda12x |
| 46 | + - name: Running dataset tests |
| 47 | + run: | |
| 48 | + [ ! -d "$HOME/verl-data" ] && git clone --depth 1 https://github.com/eric-haibin-lin/verl-data ~/verl-data |
| 49 | + python3 examples/data_preprocess/geo3k.py |
| 50 | + pytest -s -x tests/verl/utils/dataset/test_rl_dataset.py |
| 51 | + pytest -s -x tests/verl/utils/dataset/test_sft_dataset.py |
| 52 | + pytest -s -x tests/verl/utils/test_import_utils.py |
| 53 | +# pytest -s -x tests/verl/utils/dataset/test_rm_dataset.py |
| 54 | + - name: Running ray test using cupy (move it to L20 when dockerfile ready) |
| 55 | + run: | |
| 56 | + cd tests/ray |
| 57 | + pytest -s -x test_rvdz.py |
0 commit comments