Remove NumPy<2 pin

Test NumPy 1 separately

Update Python versions

Fix docs build with NumPy 2

Author: tomwhite

.github/workflows/build-numpy-2.yml renamed to .github/workflows/build-numpy-1.yml

@@ -1,4 +1,4 @@
-name: Build NumPy 2
+name: Build NumPy 1
 
 on:
   push:
@@ -11,7 +11,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.10", "3.11", "3.12"]
+        python-version: ["3.10", "3.11"]
 
     steps:
     - uses: actions/checkout@v2
@@ -22,9 +22,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install -r requirements-numpy2.txt -r requirements-numpy2-dev.txt
-        # update bio2zarr for NumPy 2, see https://github.com/sgkit-dev/bio2zarr/issues/256
-        pip install -U 'numpy<2.1' -U git+https://github.com/sgkit-dev/bio2zarr.git
+        pip install -r requirements-numpy1.txt -r requirements-numpy1-dev.txt
     # - name: Run pre-commit
     #   uses: pre-commit/[email protected]
     - name: Test with pytest and coverage

.github/workflows/build.yml

@@ -14,7 +14,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.10", "3.11"]
+        python-version: ["3.10", "3.11", "3.12"]
 
     steps:
     - uses: actions/checkout@v2

.github/workflows/check-docs.yml

@@ -14,7 +14,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.10"]
+        python-version: ["3.11"]
 
     steps:
     - uses: actions/checkout@v2
@@ -28,6 +28,7 @@ jobs:
         sudo apt install graphviz  # Needed for documentation
         python -m pip install --upgrade pip
         pip install -r requirements.txt -r requirements-dev.txt -r requirements-doc.txt
+        pip install -U dask distributed  # need latest versions to successully build docs
     - name: Run pre-commit
       uses: pre-commit/[email protected]
     - name: Check for Sphinx doc warnings

.github/workflows/docs.yml

@@ -15,13 +15,14 @@ jobs:
     - name: Set up Python ${{ matrix.python-version }}
       uses: actions/setup-python@v2
       with:
-        python-version: "3.10"
+        python-version: "3.11"
     - name: Install dependencies
       run: |
         sudo apt update -y
         sudo apt install graphviz  # Needed for documentation
         python -m pip install --upgrade pip
         pip install -r requirements.txt -r requirements-dev.txt -r requirements-doc.txt
+        pip install -U dask distributed  # need latest versions to successully build docs
     - name: Build Sphinx documentation
       run: |
         cd docs

docs/changelog.rst

@@ -29,8 +29,11 @@ Breaking changes
 .. Deprecations
 .. ~~~~~~~~~~~~
 
-.. Improvements
-.. ~~~~~~~~~~~~
+Improvements
+~~~~~~~~~~~~
+
+- Add support for NumPy 2.
+  (:user:`tomwhite`, :pr:`1297`)
 
 .. Bug fixes
 .. ~~~~~~~~~

requirements-dev.txt

@@ -12,7 +12,7 @@ scikit-learn
 partd
 bed-reader
 rechunker
-cbgen < 1.0.5
+cbgen > 1.0.5
 bio2zarr @ git+https://github.com/sgkit-dev/bio2zarr.git; platform_system != "Windows"
 yarl
 matplotlib

requirements-numpy2-dev.txt renamed to requirements-numpy1-dev.txt

@@ -12,7 +12,8 @@ scikit-learn
 partd
 bed-reader
 rechunker
-cbgen > 1.0.5
+cbgen < 1.0.5
+bio2zarr @ git+https://github.com/sgkit-dev/bio2zarr.git; platform_system != "Windows"
 yarl
 matplotlib
 asv

requirements-numpy2.txt renamed to requirements-numpy1.txt

@@ -1,4 +1,4 @@
-numpy < 2.1
+numpy < 2
 xarray < 2025.03.1
 dask[array,dataframe] >= 2023.01.0, <= 2024.8.0
 distributed >= 2023.01.0, <= 2024.8.0

requirements.txt

@@ -1,4 +1,4 @@
-numpy < 2
+numpy < 2.2
 xarray < 2025.03.1
 dask[array,dataframe] >= 2023.01.0, <= 2024.8.0
 distributed >= 2023.01.0, <= 2024.8.0

setup.cfg

@@ -27,7 +27,7 @@ zip_safe = False  # https://mypy.readthedocs.io/en/latest/installed_packages.htm
 include_package_data = True
 python_requires = >=3.10
 install_requires =
-    numpy < 2
+    numpy < 2.2
     xarray < 2025.03.1
     dask[array,dataframe] >= 2022.01.0, <= 2024.8.0
     distributed >= 2022.01.0, <= 2024.8.0
@@ -60,7 +60,7 @@ plink =
     bed-reader
 bgen =
     rechunker
-    cbgen < 1.0.5
+    cbgen > 1.0.5
 
 [coverage:report]
 fail_under = 100