Test the compatibility to numpy 2.0 (#746)

Summary: Pull Request resolved: #746

Differential Revision: D71702108

Author: HuanyuZhang

.github/workflows/ci_cpu.yml

@@ -21,7 +21,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v2
         with:
-          python-version: 3.9
+          python-version: '3.9'
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
@@ -35,32 +35,32 @@ jobs:
         run: isort -l 88 -o opacus --lines-after-imports 2 -m 3 --trailing-comma --check-only .
 
   ########### UNIT TESTS ##############
-  unittest_py38_torch_release:
+  unittest_py310_torch_release:
     runs-on: ubuntu-latest
     steps:
       - name: Checkout
         uses: actions/checkout@v2
       - name: Set up Python
         uses: actions/setup-python@v2
         with:
-          python-version: 3.8
+          python-version: '3.10'
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
           pip install pytest coverage coveralls
           ./scripts/install_via_pip.sh
       - name: Run unit tests
         run: |
-          mkdir unittest-py38-release-reports
+          mkdir unittest-py310-release-reports
           coverage run -m pytest --doctest-modules -p conftest opacus
           coverage report -i -m
           # Format into xml to be used for coveralls
           coverage xml -i
       - name: Store test results
         uses: actions/upload-artifact@v4
         with:
-          name: unittest-py38-release-reports
-          path: unittest-py38-release-reports
+          name: unittest-py310-release-reports
+          path: unittest-py310-release-reports
       - name: Send coverage to Coveralls (parallel)
         uses: coverallsapp/github-action@v2
         with:
@@ -179,7 +179,7 @@ jobs:
 
   ######## FINISH COVERALLS ##########
   finish_coveralls_parallel:
-    needs: [unittest_py38_torch_release, unittest_py39_torch_release, integrationtest_py39_torch_release_cpu]
+    needs: [unittest_py310_torch_release, unittest_py39_torch_release, integrationtest_py39_torch_release_cpu]
     runs-on: ubuntu-latest
     steps:
       - name: Checkout

.github/workflows/python-publish.yml

@@ -17,7 +17,7 @@ jobs:
     - name: Set up Python
       uses: actions/setup-python@v2
       with:
-        python-version: '3.8'
+        python-version: '3.9'
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip

.github/workflows/test-installation.yml

@@ -15,7 +15,7 @@ jobs:
       max-parallel: 48
       matrix:
         os: [windows-latest, ubuntu-latest, macos-latest]
-        python-version: ["3.8", "3.9", "3.10"]
+        python-version: ["3.9", "3.10"]
         torch-version: [1.13.1]
 
     runs-on: ${{ matrix.os }}

requirements.txt

@@ -1,4 +1,4 @@
-numpy>=1.15,<2.0
+numpy>=2.0
 torch>=2.0
 scipy>=1.2
 opt-einsum>=3.3.0