CI: add a build against numpy nightly

jax-ml · Dec 19, 2023 · 21e0cc9 · 21e0cc9
1 parent 0d1c7f6
commit 21e0cc9
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 2 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -23,17 +23,23 @@ jobs:
           python-version: 3.9
       - uses: pre-commit/action@v3.0.0
   build:
-    name: ${{ matrix.os }} Python ${{ matrix.python-version }}
+    name: ${{ matrix.os }} Python ${{ matrix.python-version }} numpy-nightly=${{ matrix.numpy-nightly }}
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
         os: ["ubuntu-latest"]
         python-version: ["3.9", "3.10", "3.11", "3.12"]
+        numpy-nightly: [false]
         include:
+          - os: ubuntu-latest
+            python-version: "3.12"
+            numpy-nightly: true
           - os: macOS-11
             python-version: "3.11"
+            numpy-nightly: false
           - os: windows-2019
             python-version: "3.11"
+            numpy-nightly: false
 
     steps:
       - uses: actions/checkout@v3
@@ -47,6 +53,12 @@ jobs:
         run: |
           python -m pip install --upgrade pip
           pip install .[dev]
+      - name: Install numpy nightly
+        if: ${{ matrix.numpy-nightly == true }}
+        run: |
+            pip install -U --pre numpy \
+              -i https://pypi.anaconda.org/scientific-python-nightly-wheels/simple
+
       - name: Run tests
         run: |
           pytest -n auto
diff --git a/ml_dtypes/tests/custom_float_test.py b/ml_dtypes/tests/custom_float_test.py
@@ -37,6 +37,14 @@
 float8_e5m2fnuz = ml_dtypes.float8_e5m2fnuz
 
 
+try:
+  # numpy >= 2.0
+  ComplexWarning = np.exceptions.ComplexWarning
+except AttributeError:
+  # numpy < 2.0
+  ComplexWarning = np.ComplexWarning
+
+
 @contextlib.contextmanager
 def ignore_warning(**kw):
   with warnings.catch_warnings():
@@ -703,7 +711,7 @@ def testCasts(self, float_type):
       self.assertTrue(np.all(x == z))
       self.assertEqual(dtype, z.dtype)
 
-  @ignore_warning(category=np.ComplexWarning)
+  @ignore_warning(category=ComplexWarning)
   def testConformNumpyComplex(self, float_type):
     for dtype in [np.complex64, np.complex128, np.clongdouble]:
       x = np.array([1.5, 2.5 + 2.0j, 3.5], dtype=dtype)

diff --git a/pytest.ini b/pytest.ini
@@ -1,3 +1,4 @@
 [pytest]
 filterwarnings =
     error
+    ignore:numpy.core._multiarray_umat.*:DeprecationWarning