pin for pytest workflow as well

Signed-off-by: Roman Lutz <[email protected]>
microsoft · Oct 10, 2023 · 4fe3253 · 4fe3253
1 parent 2255590
commit 4fe3253
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/CI-responsibleai-text-vision-pytest.yml b/.github/workflows/CI-responsibleai-text-vision-pytest.yml
@@ -57,13 +57,13 @@ jobs:
         name: Install pytorch on non-MacOS
         shell: bash -l {0}
         run: |
-          conda install --yes --quiet pytorch torchvision captum cpuonly -c pytorch
+          conda install --yes --quiet "numpy=1.21.5" "pytorch<2" torchvision captum cpuonly -c pytorch
 
       - if: ${{ matrix.operatingSystem == 'macos-latest' }}
         name: Install Anaconda packages on MacOS, which should not include cpuonly according to official docs
         shell: bash -l {0}
         run: |
-          conda install --yes --quiet pytorch torchvision captum -c pytorch
+          conda install --yes --quiet "numpy=1.21.5" "pytorch<2" torchvision captum -c pytorch
 
       - name: Setup tools
         shell: bash -l {0}