Skip to content

Commit a0b5917

Browse files
committed
make sure numpy==1.26.4 is used
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent 9cc79df commit a0b5917

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

.github/workflows/e2e-reusable.yml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ jobs:
137137

138138
- name: Install python test dependencies
139139
run: |
140-
pip install pyyaml pandas scipy 'numpy==1.26.4' psutil
140+
pip install pyyaml pandas scipy psutil
141141
142142
- name: Install transformers package
143143
if: ${{ inputs.suite == 'huggingface' }}
@@ -208,6 +208,9 @@ jobs:
208208
run: |
209209
cd pytorch
210210
211+
# Some models are still not compatible with numpy>2.0
212+
pip install 'numpy==1.26.4'
213+
211214
export WORKSPACE=$GITHUB_WORKSPACE
212215
213216
if [[ "${{ inputs.TORCH_COMPILE_DEBUG }}" = "1" ]] ; then

0 commit comments

Comments
 (0)