File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -2,11 +2,13 @@ name: Slow tests (on push)
22
33on :
44 push :
5- branches : [main]
6- paths :
7- # Run only when python files are modified
8- - " trl/**.py"
9- - " examples/**.py"
5+ branches :
6+ - main
7+ - ci-*
8+ # paths:
9+ # # Run only when python files are modified
10+ # - "trl/**.py"
11+ # - "examples/**.py"
1012env :
1113 CI_SLACK_CHANNEL : ${{ secrets.CI_PUSH_MAIN_CHANNEL }}
1214 HF_TOKEN : ${{ secrets.HF_TOKEN }}
Original file line number Diff line number Diff line change 11name : Tests (experimental)
22
33on :
4+ push :
5+ branches :
6+ - ci-*
47 pull_request :
58 paths :
69 # Run only when relevant files are modified
3841 run : pre-commit run --all-files --show-diff-on-failure --color=always
3942
4043 tests :
41- name : Tests (experimental)
44+ name : Tests
4245 runs-on :
4346 group : aws-g6e-4xlarge
4447 container :
7780 run : |
7881 source .venv/bin/activate
7982 make test_experimental
83+
84+ tests_dev :
85+ name : Tests with dev dependencies
86+ runs-on :
87+ group : aws-g6e-4xlarge
88+ container :
89+ image : pytorch/pytorch:2.8.0-cuda12.8-cudnn9-devel
90+ options : --gpus all
91+ defaults :
92+ run :
93+ shell : bash
94+ steps :
95+ - name : Git checkout
96+ uses : actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
97+
98+ - name : Set up Python 3.13
99+ uses : actions/setup-python@5fda3b95a4ea91299a34e894583c3862153e4b97 # v7.0.0
100+ with :
101+ python-version : 3.13
102+
103+ - name : Install Make and Git
104+ run : |
105+ apt-get update && apt-get install -y make git curl
106+
107+ - name : Install uv
108+ uses : astral-sh/setup-uv@c771a70e6277c0a99b617c7a806ffedaca235ff9 # v9.0.0
109+
110+ - name : Create Python virtual environment
111+ run : |
112+ uv venv
113+ uv pip install --upgrade setuptools wheel
114+
115+ - name : Install dependencies
116+ run : |
117+ source .venv/bin/activate
118+ uv pip install -U \
119+ ".[dev]" \
120+ "accelerate @ git+https://github.com/huggingface/accelerate.git" \
121+ "datasets @ git+https://github.com/huggingface/datasets.git" \
122+ "transformers @ git+https://github.com/huggingface/transformers.git" \
123+ "peft @ git+https://github.com/huggingface/peft.git"
124+
125+ - name : Test with pytest
126+ run : |
127+ source .venv/bin/activate
128+ make test_experimental
You can’t perform that action at this time.
0 commit comments