vector 1.8.0__tar.gz → 1.9.0__tar.gz

{vector-1.8.0 → vector-1.9.0}/.github/dependabot.yml

@@ -4,8 +4,10 @@ updates:
   - package-ecosystem: "github-actions"
     directory: "/"
     schedule:
-      interval: "weekly"
+      interval: "monthly"
     groups:
-      actions:
+      github-actions:
         patterns:
           - "*"
+    cooldown:
+      default-days: 7

vector-1.9.0/.github/workflows/cd.yml

@@ -0,0 +1,68 @@
+name: CD
+
+on:
+  workflow_dispatch:
+  release:
+    types:
+      - published
+
+permissions: {}
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: false
+
+jobs:
+  dist:
+    name: Build sdist and wheel
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
+        with:
+          fetch-depth: 0
+          persist-credentials: false
+
+      - uses: hynek/build-and-inspect-python-package@2abe76da66d0a6a4a227101f9348ee855797cfa5 # v3.0.1
+
+  publish:
+    name: Publish to PyPI
+    needs: [dist]
+    runs-on: ubuntu-latest
+    if: github.event_name == 'release' && github.event.action == 'published'
+    environment:
+      name: pypi
+      url: https://pypi.org/p/vector
+    permissions:
+      id-token: write # trusted publishing to PyPI
+      attestations: write # record build provenance for the sdist and the wheel
+      contents: read # gh attestation verify reads this repository
+
+    steps:
+      - uses: actions/download-artifact@3e5f45b2cfb9172054b4087a40e8e0b5a5461e7c # v8.0.1
+        with:
+          name: Packages
+          path: dist
+
+      - name: List distributions to be deployed
+        run: ls -l dist/
+
+      - name: Generate artifact attestation for sdist and wheel
+        uses: actions/attest@1e69f48acb82d1966a394da916b4c1698aa569d6 # v4.2.2
+        with:
+          subject-path: "dist/vector-*"
+
+      - name: Verify sdist artifact attestation
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          REPOSITORY: ${{ github.repository }}
+        run: gh attestation verify dist/vector-*.tar.gz --repo "$REPOSITORY"
+
+      - name: Verify wheel artifact attestation
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          REPOSITORY: ${{ github.repository }}
+        run: gh attestation verify dist/vector-*.whl --repo "$REPOSITORY"
+
+      - uses: pypa/gh-action-pypi-publish@dc37677b2e1c63e2034f94d8a5b11f265b73ba33 # v1.14.2
+        with:
+          attestations: true

vector-1.9.0/.github/workflows/ci.yml

@@ -0,0 +1,145 @@
+name: CI
+
+on:
+  pull_request:
+  push:
+    branches:
+      - main
+      - develop
+  workflow_dispatch:
+
+permissions: {}
+
+concurrency:
+  # Skip intermediate builds: always.
+  # Cancel intermediate builds: only if it is a pull request build.
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: ${{ startsWith(github.ref, 'refs/pull/') }}
+
+env:
+  # The "FORCE_COLOR" variable, when set to 1,
+  # tells Nox to colorize itself.
+  FORCE_COLOR: "1"
+
+jobs:
+  pre-commit:
+    name: Check SDist
+    runs-on: ubuntu-latest
+    permissions: {}
+    steps:
+      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
+        with:
+          persist-credentials: false
+      - uses: actions/setup-python@5fda3b95a4ea91299a34e894583c3862153e4b97 # v7.0.0
+        with:
+          python-version: 3.x
+      - uses: astral-sh/setup-uv@20cfd1bf945f4377ade1205e4dbc17946fc9a30d # v10.0.1
+      - name: PyLint
+        run: uvx nox -s pylint -- --output-format=github
+
+  check-lite:
+    runs-on: ubuntu-latest
+    permissions: {}
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version:
+          - "3.10"
+          - "3.11"
+          - "3.12"
+          - "3.13"
+          - "3.14"
+          - "3.14t"
+          - "3.15"
+          - "3.15t"
+    name: Python ${{ matrix.python-version }} - Lite
+    steps:
+      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
+        with:
+          persist-credentials: false
+
+      - uses: actions/setup-python@5fda3b95a4ea91299a34e894583c3862153e4b97 # v7.0.0
+        with:
+          python-version: ${{ matrix.python-version }}
+          allow-prereleases: true
+
+      - uses: astral-sh/setup-uv@20cfd1bf945f4377ade1205e4dbc17946fc9a30d # v10.0.1
+
+      - name: Test lite package
+        run: uvx nox -s lite-$PYTHON --verbose
+        env:
+          PYTHON: ${{ matrix.python-version }}
+
+  check-full:
+    needs: [check-lite]
+    runs-on: ubuntu-latest
+    permissions: {}
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version:
+          - "3.10"
+          - "3.11"
+          - "3.12"
+          - "3.13"
+          - "3.14"
+          - "3.14t"
+          - "3.15"
+          - "3.15t"
+    name: Python ${{ matrix.python-version }} - Full
+    steps:
+      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
+        with:
+          persist-credentials: false
+
+      - uses: actions/setup-python@5fda3b95a4ea91299a34e894583c3862153e4b97 # v7.0.0
+        with:
+          python-version: ${{ matrix.python-version }}
+          allow-prereleases: true
+
+      - uses: astral-sh/setup-uv@20cfd1bf945f4377ade1205e4dbc17946fc9a30d # v10.0.1
+
+      - name: Install nox
+        run: uv tool install nox
+
+      - name: Run doctests on Python 3.11
+        if: matrix.python-version == 3.11
+        run: nox -s doctests-$PYTHON --verbose
+        env:
+          PYTHON: ${{ matrix.python-version }}
+
+      - name: Test package and generate coverage report
+        run: nox -s coverage-$PYTHON --verbose
+        env:
+          PYTHON: ${{ matrix.python-version }}
+
+      - name: Upload coverage report
+        uses: codecov/codecov-action@fb8b3582c8e4def4969c97caa2f19720cb33a72f # v7.0.0
+        with:
+          token: ${{ secrets.CODECOV_TOKEN }}
+
+  root:
+    name: ROOT comparison tests
+    runs-on: ubuntu-latest
+    permissions: {}
+    steps:
+      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
+        with:
+          persist-credentials: false
+      - uses: prefix-dev/setup-pixi@f00437f565399d418b0acc85936d12c1fb668347 # v0.10.1
+        with:
+          environments: root
+      - name: Run ROOT comparison tests
+        run: pixi run root-tests
+
+  pass:
+    name: Pass
+    needs: [pre-commit, check-lite, check-full, root]
+    if: always()
+    runs-on: ubuntu-latest
+    permissions: {}
+    steps:
+      - name: Decide whether all required jobs succeeded
+        run: $SUCCESS
+        env:
+          SUCCESS: ${{ !contains(needs.*.result, 'failure') && !contains(needs.*.result, 'cancelled') }}

vector-1.9.0/.github/workflows/gpu.yml

@@ -0,0 +1,48 @@
+name: GPU
+
+on:
+  schedule:
+    - cron: "19 5 * * *"
+  push:
+    branches:
+      - main
+  workflow_dispatch:
+
+permissions: {}
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: ${{ startsWith(github.ref, 'refs/pull/') }}
+
+env:
+  FORCE_COLOR: "1"
+
+jobs:
+  gpu-tests:
+    name: CUDA ${{ matrix.cuda-version }}
+    if: github.event_name != 'schedule' || github.repository_owner == 'scikit-hep'
+    runs-on: self-hosted
+    permissions: {}
+    timeout-minutes: 60
+
+    strategy:
+      fail-fast: false
+      matrix:
+        cuda-version:
+          - 12
+          - 13
+
+    steps:
+      - name: Clean the workspace
+        run: rm -rf * .[!.]* || echo "Nothing to clean"
+
+      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
+        with:
+          persist-credentials: false
+
+      - uses: astral-sh/setup-uv@20cfd1bf945f4377ade1205e4dbc17946fc9a30d # v10.0.1
+
+      - name: Run the GPU tests
+        run: uvx nox -s gpu --verbose --download-python always
+        env:
+          CUDA_VERSION: ${{ matrix.cuda-version }}

{vector-1.8.0 → vector-1.9.0}/.github/workflows/notebooks.yml

@@ -7,6 +7,8 @@ on:
       - develop
   workflow_dispatch:
 
+permissions: {}
+
 concurrency:
   group: ${{ github.workflow }}-${{ github.ref }}
   cancel-in-progress: true
@@ -18,18 +20,21 @@ env:
 
 jobs:
   test:
+    name: Test notebooks
     runs-on: ubuntu-latest
+    permissions: {}
 
     steps:
-      - uses: actions/checkout@v6
+      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
+        with:
+          persist-credentials: false
 
       - name: Set up Python
-        uses: actions/setup-python@v6
+        uses: actions/setup-python@5fda3b95a4ea91299a34e894583c3862153e4b97 # v7.0.0
         with:
           python-version: "3.10"
 
-      - name: List installed Python packages
-        run: python -m pip list
+      - uses: astral-sh/setup-uv@20cfd1bf945f4377ade1205e4dbc17946fc9a30d # v10.0.1
 
       - name: Test example notebooks
-        run: pipx run nox -s notebooks
+        run: uvx nox -s notebooks --verbose

{vector-1.8.0 → vector-1.9.0}/.gitignore

@@ -148,3 +148,7 @@ uv.lock
 
 # MacOS
 .DS_Store
+
+# pixi
+/pixi.lock
+.pixi/

{vector-1.8.0 → vector-1.9.0}/.pre-commit-config.yaml

@@ -1,10 +1,11 @@
 ci:
   autoupdate_commit_msg: "chore: update pre-commit hooks"
   autofix_commit_msg: "style: pre-commit fixes"
+  autoupdate_schedule: monthly
 
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v6.0.0
+    rev: 3e8a8703264a2f4a69428a0aa4dcb512790b2c8c # frozen: v6.0.0
     hooks:
       - id: check-added-large-files
       - id: check-case-conflict
@@ -15,52 +16,60 @@ repos:
       - id: end-of-file-fixer
         exclude: ^docs
       - id: mixed-line-ending
-      - id: requirements-txt-fixer
       - id: trailing-whitespace
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: "v0.14.14"
+    rev: 321478e58f4938179c6b86e4ddfa923d1547a49b # frozen: v0.16.6
     hooks:
       - id: ruff-check
         args: ["--fix", "--show-fixes"]
       - id: ruff-format
 
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.19.1
+    rev: 7ff8d35ae36a7d2b968f2f90b4c723e292e594ee # frozen: v2.3.1
     hooks:
       - id: mypy
         files: src
         args: []
         additional_dependencies:
-          - numpy
+          # numpy>=2.5 requires Python>=3.12 and its stubs use PEP 695 syntax,
+          # which mypy cannot parse while python_version is pinned to 3.10
+          - numpy<2.5
           - packaging
           - sympy
           - optree
 
   - repo: https://github.com/codespell-project/codespell
-    rev: v2.4.1
+    rev: 57b21406f092110c18776e39b0bda50d37c945c8 # frozen: v2.4.3
     hooks:
       - id: codespell
 
   - repo: https://github.com/rbubley/mirrors-prettier
-    rev: "v3.8.1"
+    rev: "0ee178619d696787ca73d210cc191d720868c631" # frozen: v3.9.6
     hooks:
       - id: prettier
         types_or: [yaml, markdown, html, css, scss, javascript, json]
         exclude: assets/js/webapp\.js
 
   - repo: https://github.com/adamchainz/blacken-docs
-    rev: 1.20.0
+    rev: dda8db18cfc68df532abf33b185ecd12d5b7b326 # frozen: 1.20.0
     hooks:
       - id: blacken-docs
         args: ["-E"]
-        additional_dependencies: [black~=24.0]
+        additional_dependencies: [black~=26.0]
 
   - repo: https://github.com/pre-commit/pygrep-hooks
-    rev: v1.10.0
+    rev: 3a6eb0fadf60b3cccfd80bad9dbb6fae7e47b316 # frozen: v1.10.0
     hooks:
       - id: python-check-blanket-type-ignore
         exclude: ^src/vector/backends/_numba_object.py$
       - id: rst-backticks
       - id: rst-directive-colons
       - id: rst-inline-touching-normal
+
+  - repo: https://github.com/zizmorcore/zizmor-pre-commit
+    rev: cef8b8350da46d8114c7e6b7272aebdccdf193ce # frozen: v1.30.0
+    hooks:
+      - id: zizmor
+        files: "^\\.github"
+        args: [--persona=pedantic]

{vector-1.8.0 → vector-1.9.0}/CONTRIBUTING.md

@@ -146,7 +146,7 @@ pytest tests/test_notebooks.py
 
 `vector`'s documentation is mainly written in the form of [docstrings](https://peps.python.org/pep-0257) and [reStructurredText](https://docutils.sourceforge.io/docs/user/rst/quickref.html). The docstrings include the description, arguments, examples, return values, and attributes of a class or a function, and the `.rst` files enable us to render this documentation on `vector`'s documentation website.
 
-`vector` primarily uses [Sphinx](https://www.sphinx-doc.org/en/master/) for rendering documentation on its website. The configuration file (`conf.py`) for `sphinx` can be found [here](https://github.com/scikit-hep/vector/blob/main/docs/conf.py). The documentation is deployed on [https://readthedocs.io]() [here](https://vector.readthedocs.io/en/latest/).
+`vector` primarily uses [Sphinx](https://www.sphinx-doc.org/en/master/) for rendering documentation on its website. The configuration file (`conf.py`) for `sphinx` can be found [here](https://github.com/scikit-hep/vector/blob/main/docs/conf.py). The documentation is deployed on [https://readthedocs.io](<>) [here](https://vector.readthedocs.io/en/latest/).
 
 Ideally, with the addition of every new feature to `vector`, documentation should be added using comments, docstrings, and `.rst` files.
 

{vector-1.8.0 → vector-1.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: vector
-Version: 1.8.0
+Version: 1.9.0
 Summary: Vector classes and utilities
 Project-URL: Bug Tracker, https://github.com/scikit-hep/vector/issues
 Project-URL: Changelog, https://vector.readthedocs.io/en/latest/changelog.html
@@ -23,6 +23,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: 3.15
 Classifier: Programming Language :: Python :: Free Threading :: 3 - Stable
 Classifier: Topic :: Scientific/Engineering
 Classifier: Topic :: Scientific/Engineering :: Information Analysis
@@ -49,6 +50,7 @@ Description-Content-Type: text/markdown
 [![Scikit-HEP][sk-badge]][sk-link]
 
 [![Actions Status][actions-badge]][actions-link]
+[![GPU Action Status][gpu-badge]][actions-link]
 [![Documentation Status][rtd-badge]][rtd-link]
 [![pre-commit.ci status][pre-commit-badge]][pre-commit-link]
 [![codecov percentage][codecov-badge]][codecov-link]
@@ -242,6 +244,7 @@ Support for this work was provided by the National Science Foundation cooperativ
 [github-discussions-link]: https://github.com/scikit-hep/vector/discussions
 [gitter-badge]: https://badges.gitter.im/Scikit-HEP/vector.svg
 [gitter-link]: https://gitter.im/Scikit-HEP/vector?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge
+[gpu-badge]: https://github.com/scikit-hep/vector/actions/workflows/gpu.yml/badge.svg
 [joss-badge]: https://joss.theoj.org/papers/10.21105/joss.07791/status.svg
 [joss-link]: https://doi.org/10.21105/joss.07791
 [license-badge]: https://img.shields.io/badge/License-BSD_3--Clause-blue.svg

{vector-1.8.0 → vector-1.9.0}/README.md

@@ -7,6 +7,7 @@
 [![Scikit-HEP][sk-badge]][sk-link]
 
 [![Actions Status][actions-badge]][actions-link]
+[![GPU Action Status][gpu-badge]][actions-link]
 [![Documentation Status][rtd-badge]][rtd-link]
 [![pre-commit.ci status][pre-commit-badge]][pre-commit-link]
 [![codecov percentage][codecov-badge]][codecov-link]
@@ -200,6 +201,7 @@ Support for this work was provided by the National Science Foundation cooperativ
 [github-discussions-link]: https://github.com/scikit-hep/vector/discussions
 [gitter-badge]: https://badges.gitter.im/Scikit-HEP/vector.svg
 [gitter-link]: https://gitter.im/Scikit-HEP/vector?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge
+[gpu-badge]: https://github.com/scikit-hep/vector/actions/workflows/gpu.yml/badge.svg
 [joss-badge]: https://joss.theoj.org/papers/10.21105/joss.07791/status.svg
 [joss-link]: https://doi.org/10.21105/joss.07791
 [license-badge]: https://img.shields.io/badge/License-BSD_3--Clause-blue.svg

{vector-1.8.0 → vector-1.9.0}/docs/conf.py

@@ -7,7 +7,7 @@
 from __future__ import annotations
 
 import importlib.metadata
-from datetime import datetime
+from datetime import datetime, timezone
 
 # -- Path setup --------------------------------------------------------------
 # If extensions (or modules to document with autodoc) are in another directory,
@@ -22,7 +22,7 @@ from datetime import datetime
 # -- Project information -----------------------------------------------------
 
 project = "Vector"
-copyright = f"2019\u2012{datetime.now().year}"
+copyright = f"2019\u2012{datetime.now(tz=timezone.utc).year}"
 author = (
     "Saransh Chopra, Henry Schreiner, Jim Pivarski, Eduardo Rodrigues, and Jonas Eschle"
 )
@@ -42,6 +42,7 @@ extensions = [
     "sphinx.ext.napoleon",
     "sphinx_copybutton",
     "sphinx_math_dollar",
+    "sphinx_llm.txt",
 ]
 
 source_suffix = [".rst", ".md"]
@@ -64,6 +65,14 @@ exclude_patterns = ["_build", "Thumbs.db", ".DS_Store"]
 
 autodoc_mock_imports = ["numba"]
 
+# -- Options for LLM-friendly output -----------------------------------------
+
+# The default is the full README, which is too long for the summary block
+llms_txt_description = (
+    "Documentation of the Vector library for 2D, 3D, and Lorentz vectors,"
+    " including the object, NumPy, Awkward, SymPy, and Numba backends."
+)
+
 # -- Options for HTML output -------------------------------------------------
 
 # The theme to use for HTML and HTML Help pages.  See the documentation for

{vector-1.8.0 → vector-1.9.0}/docs/index.md

@@ -1,3 +1,11 @@
+---
+myst:
+  html_meta:
+    description:
+      Vector is a Python library for 2D, 3D, and Lorentz vectors, with
+      object, NumPy, Awkward, SymPy, and Numba backends.
+---
+
 ![](_images/LogoSrc.svg)
 
 # Overview

{vector-1.8.0 → vector-1.9.0}/docs/src/make_awkward.md

@@ -27,7 +27,18 @@ and thus the `abs` function computes the magnitude of each record as `sqrt(x**2
 
 It is not necessary to install Vector's behaviors globally. They could be installed in the `vec` array only by passing `behavior=vector.backends.awkward.behavior` to the [ak.Array](https://awkward-array.org/doc/main/reference/generated/ak.Array.html) constructor.
 
-The records can contain more fields than those that specify coordinates, which can be useful for specifying properties of a particle other than its momentum. Only the coordinate names are considered when performing vector calculations. Coordinates must be numbers (not, for instance, lists of numbers). Be careful about field names that coincide with coordinates, such as `rho` (azimuthal magnitude) and `tau` (proper time).
+The records can contain more fields than those that specify coordinates, which can be useful for specifying properties of a particle other than its momentum. Only the coordinate names are considered when performing vector calculations. Coordinates must be numbers (not, for instance, lists of numbers).
+
+Field names that mean a coordinate under another name, such as `rho` (a synonym of `pt`) or `energy` in a record that already has `mass`, would silently change what a vector means, so Vector rejects them:
+
+```python
+>>> ak.Array([{"pt": 1.1, "phi": 2.2, "eta": 3.3, "mass": 4.4, "energy": 5.5}], with_name="Momentum4D")
+Traceback (most recent call last):
+    ...
+TypeError: MomentumArray4D with fields ['pt', 'phi', 'eta', 'mass', 'energy']: specify t= or tau=, but not more than one (got 'energy', 'mass')
+```
+
+This runs whenever the behaviors are attached, not only in `vector.Array` and `vector.zip`, so `jets["rho"] = ...` is caught too. It is the `__awkward_validation__` hook of each behavior class and needs `awkward>=2.8.11`; older versions skip it. A rejected in-place assignment is only undone by `awkward>=2.12`: before that, the `TypeError` is raised but the array keeps the offending field. Subclassed behaviors (see [Advanced: subclassing Awkward-Vector behaviors](awkward.ipynb)) inherit the check and can extend it by overriding `__awkward_validation__` and calling `super().__awkward_validation__()`.
 
 The `vector.Array` function (`vector.awk` is a synonym) is an alternative to the [ak.Array](https://awkward-array.org/doc/main/reference/generated/ak.Array.html) constructor, which installs Vector's behavior in the new array (not globally in `ak.behavior`).
 

{vector-1.8.0 → vector-1.9.0}/environment.yml

@@ -5,12 +5,12 @@ dependencies:
   - jupyterlab >=1.2
   - nb_conda_kernels
   - pip >=18
-  - pytest >=6
+  - pytest >=7
   - numba >=0.57
   - numpy >=1.19.3
   - root >=6.18.04
   - pip:
       - "awkward>=2.0.0"
       - "uproot>=5.0.0"
-      - "scikit-hep-testdata>=0.2.0"
+      - "scikit-hep-testdata>=0.6.0"
       - -e .

{vector-1.8.0 → vector-1.9.0}/noxfile.py

@@ -7,8 +7,6 @@
 from __future__ import annotations
 
 import os
-import sys
-import sysconfig
 from pathlib import Path
 
 import nox
@@ -19,7 +17,7 @@ nox.options.default_venv_backend = "uv|virtualenv"
 DIR = Path(__file__).parent.resolve()
 PYPROJECT = nox.project.load_toml(DIR / "pyproject.toml")
 ALL_PYTHON = nox.project.python_versions(PYPROJECT)
-ALL_PYTHON += ["3.14t"]  # add free-threaded Python variant
+ALL_PYTHON += ["3.14t", "3.15t"]  # add free-threaded Python variant
 
 
 @nox.session(reuse_venv=True)
@@ -40,27 +38,32 @@ def pylint(session: nox.Session) -> None:
 @nox.session(reuse_venv=True, python=ALL_PYTHON)
 def lite(session: nox.Session) -> None:
     """Run lightweight tests."""
+    is_free_threaded = isinstance(session.python, str) and session.python.endswith("t")
+    run_env = {"PYTHON_GIL": "0"} if is_free_threaded else {}
     test_deps = nox.project.dependency_groups(PYPROJECT, "test")
     session.install("-e.", *test_deps)
-    session.run("pytest", "--ignore", "tests/test_notebooks.py", *session.posargs)
+    session.run(
+        "pytest",
+        "--ignore",
+        "tests/test_notebooks.py",
+        *session.posargs,
+        env=run_env,
+    )
 
 
 @nox.session(reuse_venv=True, python=ALL_PYTHON)
 def tests(session: nox.Session) -> None:
     """Run the unit and regular tests."""
-    if sys.version_info[:2] >= (3, 14) and bool(
-        sysconfig.get_config_var("Py_GIL_DISABLED")
-    ):
-        os.environ["PYTHON_GIL"] = "0"
-        test_deps = nox.project.dependency_groups(PYPROJECT, "test-all-no-gil")
-    else:
-        test_deps = nox.project.dependency_groups(PYPROJECT, "test-all-gil")
+    is_free_threaded = isinstance(session.python, str) and session.python.endswith("t")
+    run_env = {"PYTHON_GIL": "0"} if is_free_threaded else {}
+    test_deps = nox.project.dependency_groups(PYPROJECT, "test-all")
     session.install("-e.", *test_deps)
     session.run(
         "pytest",
         "--ignore",
         "tests/test_notebooks.py",
         *session.posargs,
+        env=run_env,
     )
 
 
@@ -74,7 +77,7 @@ def coverage(session: nox.Session) -> None:
 @nox.session(reuse_venv=True, python=ALL_PYTHON)
 def doctests(session: nox.Session) -> None:
     """Run the doctests."""
-    test_deps = nox.project.dependency_groups(PYPROJECT, "test-all-gil")
+    test_deps = nox.project.dependency_groups(PYPROJECT, "test-all")
     session.install("-e.", *test_deps)
     session.run("pytest", "--doctest-plus", "src/vector/", *session.posargs)
 
@@ -82,19 +85,30 @@ def doctests(session: nox.Session) -> None:
 @nox.session(reuse_venv=True, default=False)
 def notebooks(session: nox.Session) -> None:
     """Run the notebook tests"""
-    test_deps = nox.project.dependency_groups(
-        PYPROJECT, "test", "test-optional", "test-numba"
-    )
+    test_deps = nox.project.dependency_groups(PYPROJECT, "test", "test-optional")
     session.install("-e.", *test_deps)
     session.install("jupyter")
     session.run("pytest", "tests/test_notebooks.py", *session.posargs)
 
 
+# Not reuse_venv: the CuPy wheel is chosen by CUDA_VERSION, so a reused venv would
+# keep the one installed for whichever version ran first.
+@nox.session(python="3.13", default=False)
+def gpu(session: nox.Session) -> None:
+    """Run the GPU tests. CUDA_VERSION picks the CuPy wheel (CUDA major, default 13)."""
+    cuda_version = os.environ.get("CUDA_VERSION", "13")
+    test_deps = nox.project.dependency_groups(PYPROJECT, "test", "test-optional")
+    # The "ctk" extra brings the CUDA runtime with it; the bare wheel expects a
+    # toolkit to be installed already. The driver is the host's either way.
+    session.install("-e.", *test_deps, f"cupy-cuda{cuda_version}x[ctk]>=14")
+    session.run("pytest", "tests/cuda", *session.posargs)
+
+
 @nox.session(reuse_venv=True, default=False)
 def docs(session: nox.Session) -> None:
     """Build the docs. Pass "serve" to serve."""
     doc_deps = nox.project.dependency_groups(PYPROJECT, "docs", "test")
-    session.install("-e.", doc_deps)
+    session.install("-e.", *doc_deps)
     session.chdir("docs")
     session.run("sphinx-build", "-M", "html", ".", "_build")