DEV: add default environments for tasks

.github/workflows/ci.yml

@@ -31,7 +31,7 @@ jobs:
 
       - uses: prefix-dev/setup-pixi@1b2de7f3351f171c8b4dfeb558c639cb58ed4ec0 # v0.9.5
         with:
-          pixi-version: v0.62.2
+          pixi-version: v0.67.2
           cache: true
           environments: lint
 
@@ -65,7 +65,7 @@ jobs:
 
       - uses: prefix-dev/setup-pixi@1b2de7f3351f171c8b4dfeb558c639cb58ed4ec0 # v0.9.5
         with:
-          pixi-version: v0.62.2
+          pixi-version: v0.67.2
           cache: true
           environments: ${{ matrix.environment }}
 

.github/workflows/docs.yml

@@ -23,7 +23,7 @@ jobs:
 
       - uses: prefix-dev/setup-pixi@1b2de7f3351f171c8b4dfeb558c639cb58ed4ec0 # v0.9.5
         with:
-          pixi-version: v0.62.0
+          pixi-version: v0.67.2
           cache: true
           environments: docs
 

pixi.toml

@@ -0,0 +1,203 @@
+[workspace]
+channels = ["https://prefix.dev/conda-forge"]
+platforms = ["linux-64", "osx-64", "osx-arm64", "win-64"]
+preview = ["pixi-build"]
+requires-pixi = ">=0.63.0"
+
+### array-api-extra package definition ###
+
+[package.build.backend]
+name = "pixi-build-python"
+version = "*"
+
+[package.host-dependencies]
+meson-python = "*"
+uv = "*" # interfaces with meson-python instead of pip
+
+[package.run-dependencies]
+array-api-compat = "*"
+
+### workspace environments ###
+
+[environments]
+default = { features = ["py314"], solve-group = "py314" }
+lint = { features = ["py314", "lint"], solve-group = "py314" }
+docs = { features = ["py314", "docs"], solve-group = "py314" }
+tests = { features = ["py314", "tests"], solve-group = "py314" }
+tests-py314 = { features = ["py314", "tests"], solve-group = "py314" } # alias of tests
+
+# Some backends may pin numpy; use separate solve-group
+dev = { features = ["py314", "lint", "tests", "docs", "dev", "backends"], solve-group = "backends" }
+tests-backends = { features = ["py314", "tests", "backends"], solve-group = "backends" }
+tests-backends-py311 = { features = ["py311", "tests", "backends"] }
+
+# CUDA not available on free github actions and on some developers' PCs
+dev-cuda = { features = ["py314", "lint", "tests", "docs", "dev", "backends", "cuda-backends"], solve-group = "cuda" }
+tests-cuda = { features = ["py314", "tests", "backends", "cuda-backends"], solve-group = "cuda" }
+tests-cuda-py311 = { features = ["py311", "tests", "backends", "cuda-backends"] }
+
+# Ungrouped environments
+tests-numpy1 = ["py311", "tests", "numpy1"]
+tests-py311 = ["py311", "tests"]
+tests-nogil = ["nogil", "tests"]
+
+### default feature definition ###
+
+[dev]
+# this pulls in array-api-extra's host and run dependencies
+array-api-extra.path = "."
+
+[dependencies]
+array-api-extra.path = "."
+
+### non-default feature definitions ###
+
+[feature.lint.dependencies]
+typing-extensions = ">=4.15.0"
+pylint = ">=4.0.5"
+mypy = ">=1.20.0"
+basedpyright = ">=1.39.0"
+numpydoc = ">=1.10.0,<2"
+# import dependencies for mypy:
+array-api-strict = ">=2.5,<2.6"
+numpy = ">=2.1.3"
+hypothesis = ">=6.151.9"
+dask-core = ">=2026.3.0" # No distributed, tornado, etc.
+dprint = ">=0.50.0,<0.51"
+lefthook = ">=2.1.5,<3"
+ruff = ">=0.15.9,<0.16"
+typos = ">=1.44.0,<2"
+actionlint = ">=1.7.12,<2"
+blacken-docs = ">=1.20.0,<2"
+pytest = ">=9.0.3,<10"
+validate-pyproject = ">=0.25,<0.26"
+pyrefly = ">=0.61.1,<0.62"
+zizmor = ">=1.24.1,<1.25"
+# NOTE: don't add cupy, jax, pytorch, or sparse here,
+# as they slow down mypy and are not portable across target OSs
+
+[feature.lint.tasks]
+lefthook = { cmd = "lefthook", description = "Run lefthook", default-environment = "lint" }
+hooks = { cmd = "lefthook install", description = "Install pre-commit hooks", default-environment = "lint" }
+pre-commit = { cmd = "lefthook run pre-commit", description = "Run pre-commit checks", default-environment = "lint" }
+pylint = { cmd = "pylint array_api_extra", cwd = "src", description = "Lint with pylint", default-environment = "lint" }
+mypy = { cmd = "mypy", description = "Type check with mypy", default-environment = "lint" }
+pyrefly = { cmd = "pyrefly check", description = "Type check with pyrefly", default-environment = "lint" }
+pyright = { cmd = "basedpyright", description = "Type check with basedpyright", default-environment = "lint" }
+ruff-check = { cmd = "ruff check --fix", description = "Lint with ruff", default-environment = "lint" }
+ruff-format = { cmd = "ruff format", description = "Format with ruff", default-environment = "lint" }
+dprint = { cmd = "dprint fmt", description = "Format with dprint", default-environment = "lint" }
+typos = { cmd = "typos --write-changes --force-exclude", description = "Fix typos", default-environment = "lint" }
+actionlint = { cmd = "actionlint", description = "Lint actions with actionlint", default-environment = "lint" }
+zizmor = { cmd = "zizmor .github -p", description = "GHA static analysis with zizmor", default-environment = "lint" }
+blacken-docs = { cmd = "blacken-docs", description = "Format Python markdown blocks with Black", default-environment = "lint" }
+validate-pyproject = { cmd = "validate-pyproject pyproject.toml", description = "Validate pyproject.toml", default-environment = "lint" }
+numpydoc = { cmd = "numpydoc lint", description = "Validate docstrings with numpydoc", default-environment = "lint" }
+lint = { cmd = "lefthook run pre-commit --all-files --force", description = "Run all linters", default-environment = "lint" }
+
+[feature.tests.dependencies]
+pytest = ">=9.0.3"
+pytest-cov = ">=7.1.0"
+hypothesis = ">=6.151.9"
+array-api-strict = ">=2.5,<2.6"
+numpy = ">=1.22.0"
+scipy = ">=1.15.2,<2"
+
+[feature.tests.tasks]
+tests = { cmd = "pytest -v", description = "Run tests", default-environment = "tests" }
+tests-cov = { cmd = "pytest -v -ra --cov --cov-report=xml --cov-report=term --durations=20", description = "Run tests with coverage", default-environment = "tests" }
+
+clean-vendor-compat = { cmd = "rm -rf vendor_tests/array_api_compat", description = "Delete the existing vendored version of array-api-compat", default-environment = "tests" }
+clean-vendor-extra = { cmd = "rm -rf vendor_tests/array_api_extra", description = "Delete the existing vendored version of array-api-extra", default-environment = "tests" }
+copy-vendor-compat = { cmd = "cp -r $(python -c 'import site; print(site.getsitepackages()[0])')/array_api_compat vendor_tests/", depends-on = ["clean-vendor-compat"], description = "Vendor a clean copy of array-api-compat", default-environment = "tests" }
+copy-vendor-extra = { cmd = "cp -r src/array_api_extra vendor_tests/", depends-on = ["clean-vendor-extra"], description = "Vendor a clean copy of array-api-extra", default-environment = "tests" }
+tests-vendor = { cmd = "pytest -v vendor_tests", depends-on = ["copy-vendor-compat", "copy-vendor-extra"], description = "Check that array-api-extra and array-api-compat can be vendored together", default-environment = "tests" }
+
+tests-ci = { depends-on = ["tests-cov", "tests-vendor"], description = "Run tests with coverage and vendor tests" }
+coverage = { cmd = "coverage html", depends-on = ["tests-cov"], description = "Generate test coverage html report", default-environment = "tests" }
+open-coverage = { cmd = "open htmlcov/index.html", depends-on = ["coverage"], description = "Open test coverage report", default-environment = "tests" }
+
+[feature.docs.dependencies]
+sphinx = ">=7.4.7"
+furo = ">=2025.12.19"
+myst-parser = ">=5.0.0"
+sphinx-copybutton = ">=0.5.2"
+sphinx-autodoc-typehints = ">=1.25.3"
+# Needed to import parsed modules with autodoc
+dask-core = ">=2026.3.0" # No distributed, tornado, etc.
+pytest = ">=9.0.3"
+typing-extensions = ">=4.15.0"
+numpy = ">=2.1.3"
+
+[feature.docs.tasks]
+docs = { cmd = "sphinx-build -E -W . build/", cwd = "docs", description = "Build docs", default-environment = "docs" }
+open-docs = { cmd = "open build/index.html", cwd = "docs", depends-on = ["docs"], description = "Open the generated docs", default-environment = "docs" }
+
+[feature.dev.dependencies]
+ipython = ">=7.33.0"
+
+[feature.dev.tasks]
+ipython = { cmd = "ipython", description = "Launch ipython", default-environment = "dev" }
+
+[feature.py311.dependencies]
+python = "~=3.11.0"
+
+[feature.py314.dependencies]
+python = "~=3.14.0"
+
+[feature.numpy1.dependencies]
+# Oldest NumPy version supported by scikit-learn.
+# Note that this is older than what SPEC0 recommends.
+numpy = "=1.24.1"
+
+# Backends that can run on CPU-only hosts
+# Note: JAX and PyTorch will install CPU variants.
+[feature.backends.dependencies]
+pytorch = ">=2.10.0"
+dask-core = ">=2026.3.0" # No distributed, tornado, etc.
+sparse = ">=0.18.0"
+
+[feature.backends.target.linux-64.dependencies]
+jax = ">=0.9.2"
+
+[feature.backends.target.osx-64.dependencies]
+jax = ">=0.9.2"
+
+[feature.backends.target.osx-arm64.dependencies]
+jax = ">=0.9.2"
+
+[feature.backends.target.win-64.dependencies]
+# jax = "*"  # unavailable
+
+# Backends that require a GPU host and a CUDA driver.
+# Note that JAX and PyTorch automatically prefer CUDA variants
+# thanks to the `system-requirements` below, *if available*.
+# We request them explicitly below to ensure that we don't
+# quietly revert to CPU-only in the future, e.g. when CUDA 13
+# is released and CUDA 12 builds are dropped upstream.
+[feature.cuda-backends]
+system-requirements = { cuda = "12" }
+
+[feature.cuda-backends.target.linux.dependencies]
+cupy = ">=14.0.1"
+jaxlib = { version = ">=0.9.2", build = "cuda12*" }
+pytorch = { version = ">=2.10.0", build = "cuda12*" }
+
+[feature.cuda-backends.target.osx.dependencies]
+# cupy = "*"  # unavailable
+# jaxlib = { version = "*", build = "cuda12*" }  # unavailable
+# pytorch = { version = "*", build = "cuda12*" }  # unavailable
+
+[feature.cuda-backends.target.win.dependencies]
+cupy = ">=14.0.1"
+# jaxlib = { version = "*", build = "cuda12*" }  # unavailable
+pytorch = { version = ">=2.10.0", build = "cuda12*" }
+
+[feature.nogil.dependencies]
+python-freethreading = "~=3.13.0"
+pytest-run-parallel = ">=0.8.2"
+numpy = ">=2.3.5"
+# pytorch = "*"  # Not available on Python 3.13t yet
+dask-core = ">=2026.3.0" # No distributed, tornado, etc.
+# sparse = "*"  # numba not available on Python 3.13t yet
+# jax = "*"  # ml_dtypes not available on Python 3.13t yet