saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-30 03:37:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[ROCm] Fix CI pipeline by fixing pytest version (#19407)

Browse source 
Fix pytest version to 7.4.4, higher version will cause error

`from onnxruntime.capi import onnxruntime_validation 
ModuleNotFoundError: No module named 'onnxruntime.capi'`
This commit is contained in:
PeixuanZuo 2024-02-04 16:37:36 +08:00 committed by GitHub
parent debd1cab10
commit 0cba56e0a0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tools/ci_build/github/pai/rocm-ci-pipeline-env.Dockerfile
View file

@ -125,7 +125,8 @@ RUN pip install \
pytorch_lightning==1.6.0 \
pytest-xdist \
pytest-rerunfailures \
ml_dtypes==0.3.0
ml_dtypes==0.3.0 \
pytest==7.4.4
# Install migraphx
RUN apt update && apt install -y migraphx