From 5bba24b4c7a6e27c1f8bd5cdb3ec76ab1dcd29af Mon Sep 17 00:00:00 2001 From: rui-ren Date: Wed, 14 Feb 2024 23:36:31 +0000 Subject: [PATCH] fix rocm ci pipeline --- tools/ci_build/github/pai/rocm-ci-pipeline-env.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/ci_build/github/pai/rocm-ci-pipeline-env.Dockerfile b/tools/ci_build/github/pai/rocm-ci-pipeline-env.Dockerfile index 64710a982a29d..496b57b417fbd 100644 --- a/tools/ci_build/github/pai/rocm-ci-pipeline-env.Dockerfile +++ b/tools/ci_build/github/pai/rocm-ci-pipeline-env.Dockerfile @@ -112,7 +112,7 @@ RUN pip install \ cerberus \ sympy \ h5py \ - datasets==1.9.0 \ + datasets==2.17.0 \ requests \ sacrebleu==1.5.1 \ sacremoses \