diff --git a/.github/unittest/windows_optdepts/scripts/environment.yml b/.github/unittest/windows_optdepts/scripts/environment.yml index 260b6f48605..2740c77f434 100644 --- a/.github/unittest/windows_optdepts/scripts/environment.yml +++ b/.github/unittest/windows_optdepts/scripts/environment.yml @@ -16,4 +16,3 @@ dependencies: - pyyaml - scipy - coverage - - numpy<2.0 diff --git a/.github/unittest/windows_optdepts/scripts/install.sh b/.github/unittest/windows_optdepts/scripts/install.sh index 5c425d18a95..f13b83a0be0 100644 --- a/.github/unittest/windows_optdepts/scripts/install.sh +++ b/.github/unittest/windows_optdepts/scripts/install.sh @@ -37,6 +37,7 @@ fi # submodules git submodule sync && git submodule update --init --recursive +python -m pip install "numpy<2.0" printf "Installing PyTorch with %s\n" "${cudatoolkit}" if [[ "$TORCH_VERSION" == "nightly" ]]; then