Skip to content

Commit

Permalink
[ROCm] Fix CI pipeline by fixing pytest version (#19407)
Browse files Browse the repository at this point in the history
Fix pytest version to 7.4.4, higher version will cause error

`from onnxruntime.capi import onnxruntime_validation 
ModuleNotFoundError: No module named 'onnxruntime.capi'`
  • Loading branch information
PeixuanZuo committed Feb 4, 2024
1 parent debd1ca commit 0cba56e
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tools/ci_build/github/pai/rocm-ci-pipeline-env.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 0cba56e

Please sign in to comment.