patch

Author: Borda

.github/workflows/ci-tests-fabric.yml

@@ -85,11 +85,9 @@ jobs:
       - name: Append Env. vars for Linux
         if: ${{ runner.os == 'Linux' }}
         run: echo "GLOO_SOCKET_IFNAME=eth0" >> $GITHUB_ENV
-
       - name: Append Env. vars for MacOS
         if: ${{ runner.os == 'macOS' }}
         run: echo "GLOO_SOCKET_IFNAME=lo0" >> $GITHUB_ENV
-
       - name: Append Env. vars for Windows
         if: ${{ runner.os == 'windows' }}
         run: |
@@ -134,6 +132,11 @@ jobs:
             --find-links="${TORCH_URL}"
           uv pip list
 
+      - name: Patch NumPy for old Torch versions
+        if: ${{ startsWith(matrix.pytorch-version, '2.1') || startsWith(matrix.pytorch-version, '2.2') || startsWith(matrix.pytorch-version, '2.3') }}
+        # Patch numpy to avoid "RuntimeError: Numpy is not available"
+        run: uv pip install -U "numpy<1.24"
+
       - name: Dump handy wheels
         if: github.event_name == 'push' && github.ref == 'refs/heads/master'
         continue-on-error: true

.github/workflows/ci-tests-pytorch.yml

@@ -138,6 +138,11 @@ jobs:
             --find-links="https://download.pytorch.org/whl/torch-tensorrt"
           uv pip list
 
+      - name: Patch NumPy for old Torch versions
+        if: ${{ startsWith(matrix.pytorch-version, '2.1') || startsWith(matrix.pytorch-version, '2.2') || startsWith(matrix.pytorch-version, '2.3') }}
+        # Patch numpy to avoid "RuntimeError: Numpy is not available"
+        run: uv pip install -U "numpy<1.24"
+
       - name: Drop LAI from extensions
         if: ${{ matrix.config.pkg-name != 'lightning' }}
         # Lightning is dependency of Habana or other accelerators/integrations so in case we test PL we need to remove it

requirements/fabric/test.txt

@@ -1,5 +1,4 @@
 coverage ==7.10.6
-numpy >=1.21.0, <2.4.0
 pytest ==8.4.2
 pytest-cov ==6.3.0
 pytest-timeout ==2.4.0

requirements/pytorch/test.txt

@@ -8,7 +8,6 @@ pytest-random-order ==1.2.0
 # needed in tests
 cloudpickle >=1.3, <3.2.0
 scikit-learn >0.22.1, <1.8.0
-numpy >1.20.0, <2.4.0
 onnx >1.12.0, <1.20.0
 onnxruntime >=1.12.0, <1.23.0
 onnxscript >= 0.1.0, < 0.5.0