diff --git a/.github/workflows/regression_test.yml b/.github/workflows/regression_test.yml index f1188fd7d5..f4f054b307 100644 --- a/.github/workflows/regression_test.yml +++ b/.github/workflows/regression_test.yml @@ -64,7 +64,7 @@ jobs: torch-spec: 'torch==2.5.1 --index-url https://download.pytorch.org/whl/cu121' gpu-arch-type: "cuda" gpu-arch-version: "12.6" - dev-requirements-overrides: "s/^pytest$/pytest==7.4.0/" + dev-requirements-overrides: "s/^pytest.*$/pytest==7.4.0/" - name: CUDA 2.6 runs-on: linux.g5.12xlarge.nvidia.gpu torch-spec: 'torch==2.6.0' @@ -83,7 +83,7 @@ jobs: torch-spec: 'torch==2.5.1 --index-url https://download.pytorch.org/whl/cpu' gpu-arch-type: "cpu" gpu-arch-version: "" - dev-requirements-overrides: "s/^pytest$/pytest==7.4.0/" + dev-requirements-overrides: "s/^pytest.*$/pytest==7.4.0/" - name: CPU 2.6 runs-on: linux.4xlarge torch-spec: 'torch==2.6.0 --index-url https://download.pytorch.org/whl/cpu' diff --git a/dev-requirements.txt b/dev-requirements.txt index 600d5001cf..f9e4381e4e 100644 --- a/dev-requirements.txt +++ b/dev-requirements.txt @@ -1,5 +1,5 @@ # Test utilities -pytest +pytest==8.3.4 unittest-xml-reporting parameterized packaging