Merge branch 'main' into dev-new-engine

Author: ValerianRey

.github/workflows/build-deploy-docs.yml

@@ -6,6 +6,10 @@ on:
     tags:
       - 'v[0-9]*.[0-9]*.[0-9]*'
 
+env:
+  UV_NO_SYNC: 1
+  PYTHON_VERSION: '3.14'
+
 jobs:
   build-deploy-doc:
     name: Build & deploy doc
@@ -20,10 +24,10 @@ jobs:
       - name: Set up uv
         uses: astral-sh/setup-uv@v5
         with:
-          python-version: '3.13'
+          python-version: ${{ env.PYTHON_VERSION }}
 
       - name: Install dependencies (default with full options & doc)
-        run: uv pip install '.[full]' --group doc
+        run: uv pip install --python-version=${{ env.PYTHON_VERSION }} '.[full]' --group doc
 
       - name: Determine deployment folder
         id: deploy_folder

.github/workflows/release.yml

@@ -4,6 +4,9 @@ on:
   release:
     types: [published]
 
+env:
+  PYTHON_VERSION: 3.14
+
 jobs:
   pypi-publish:
     name: Publish to PyPI
@@ -19,7 +22,7 @@ jobs:
       - name: Set up uv
         uses: astral-sh/setup-uv@v5
         with:
-          python-version: '3.13'
+          python-version: ${{ env.PYTHON_VERSION }}
 
       - name: Build
         run: uv build

.github/workflows/tests.yml

@@ -6,14 +6,18 @@ on:
   schedule:
     - cron: '41 16 * * *'  # Every day at 16:41 UTC (to avoid high load at exact hour values).
 
+env:
+  UV_NO_SYNC: 1
+  PYTHON_VERSION: 3.14
+
 jobs:
   tests-full-install:
     name: Run tests with full install
     runs-on: ${{ matrix.os }}
     strategy:
       fail-fast: false  # Ensure matrix jobs keep running even if one fails
       matrix:
-        python-version: ['3.10', '3.11', '3.12', '3.13']
+        python-version: ['3.10', '3.11', '3.12', '3.13', '3.14']
         os: [ubuntu-latest, macOS-latest, windows-latest]
 
     steps:
@@ -23,7 +27,7 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
       - name: Install default (with full options) and test dependencies
-        run: uv pip install '.[full]' --group test
+        run: uv pip install --python-version=${{ matrix.python-version }} '.[full]' --group test
       - name: Run unit and doc tests with coverage report
         run: uv run pytest -W error tests/unit tests/doc --cov=src --cov-report=xml
       - name: Upload results to Codecov
@@ -39,9 +43,9 @@ jobs:
       - name: Set up uv
         uses: astral-sh/setup-uv@v5
         with:
-          python-version: '3.13'
+          python-version: ${{ env.PYTHON_VERSION }}
       - name: Install default (without any option) and test dependencies
-        run: uv pip install . --group test
+        run: uv pip install --python-version=${{ env.PYTHON_VERSION }} . --group test
       - name: Run unit and doc tests with coverage report
         run: |
           uv run pytest -W error tests/unit tests/doc \
@@ -64,10 +68,10 @@ jobs:
       - name: Set up uv
         uses: astral-sh/setup-uv@v5
         with:
-          python-version: '3.13'
+          python-version: ${{ env.PYTHON_VERSION }}
 
       - name: Install dependencies (default with full options & doc)
-        run: uv pip install '.[full]' --group doc
+        run: uv pip install --python-version=${{ env.PYTHON_VERSION }} '.[full]' --group doc
 
       - name: Build Documentation
         working-directory: docs
@@ -83,10 +87,10 @@ jobs:
     - name: Set up uv
       uses: astral-sh/setup-uv@v5
       with:
-        python-version: '3.13'
+        python-version: ${{ env.PYTHON_VERSION }}
 
     - name: Install dependencies (default with full options & check)
-      run: uv pip install '.[full]' --group check
+      run: uv pip install --python-version=${{ env.PYTHON_VERSION }} '.[full]' --group check
 
     - name: Run mypy
       run: uv run mypy src/torchjd

CHANGELOG.md

@@ -35,6 +35,7 @@ changes that do not affect the user.
 - Added usage example for IWRM with autogram.
 - Added usage example for IWRM with partial autogram.
 - Added usage example for IWMTL with autogram.
+- Added Python 3.14 classifier in pyproject.toml (we now also run tests on Python 3.14 in the CI).
 
 ### Changed
 

CONTRIBUTING.md

@@ -12,14 +12,28 @@ mandatory, we only provide installation steps with this tool. You can install it
 1) Pre-requisites: Use `uv` to install a Python version compatible with TorchJD and to pin it to the
   `torchjd` folder. From the root of the `torchjd` repo, run:
    ```bash
-   uv python install 3.13.3
-   uv python pin 3.13.3
+   uv python install 3.14.0
+   uv python pin 3.14.0
    ```
 
 2) Create a virtual environment and install the project in it. From the root of `torchjd`, run:
    ```bash
    uv venv
-   CC=gcc uv pip install -e '.[full]' --group check --group doc --group test --group plot
+   CC=gcc uv pip install --python-version=3.13.3 -e '.[full]' --group check --group doc --group test --group plot
+   ```
+   We also advise using `UV_NO_SYNC=1` to prevent `uv` from syncing all the time. This is because by
+   default, it tries to resolve libraries compatible with the whole range of Python versions
+   supported by TorchJD, but in reality, we just need an installation compatible with the currently
+   used Python version. That's also why we specify `--python-version=3.14` when running
+   `uv pip install`. To follow that recommendation, add the following line to your `.bashrc`:
+   ```bash
+   export UV_NO_SYNC=1
+   ```
+   and start a new terminal. The alternative is to use the `--no-sync` flag whenever you run a pip
+   command that would normally sync (like `uv run`).
+
+3) Install pre-commit:
+   ```bash
    uv run pre-commit install
    ```
 
@@ -46,7 +60,7 @@ from the root of `torchjd`:
 rm -rf .venv
 rm uv.lock
 uv venv
-CC=gcc uv pip install -e '.[full]' --group check --group doc --group test --group plot
+CC=gcc uv pip install --python-version=3.13.3 -e '.[full]' --group check --group doc --group test --group plot
 uv run pre-commit install
 ```
 

docs/source/installation.md

@@ -5,7 +5,7 @@
 :end-before: <!-- end installation -->
 ```
 
-Note that `torchjd` requires Python 3.10, 3.11, 3.12 or 3.13 and `torch>=2.0`.
+Note that `torchjd` requires Python 3.10, 3.11, 3.12, 3.13 or 3.14 and `torch>=2.0`.
 
 Some aggregators (CAGrad and Nash-MTL) have additional dependencies that are not included by default
 when installing `torchjd`. To install them, you can use:

pyproject.toml

@@ -29,6 +29,7 @@ classifiers = [
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
     "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
     "Topic :: Scientific/Engineering",
     "Topic :: Scientific/Engineering :: Artificial Intelligence",
 ]
@@ -71,9 +72,9 @@ check = [
 
 doc = [
     "sphinx>=6.0, !=7.2.0, !=7.2.1, !=7.2.3, !=7.2.4, !=7.2.5",  # Versions in [7.2.0, 7.2.5] have a bug with an internal torch import from _C
-    "furo>=2023.0, <2024.04.27",  # Force it to be recent so that the theme looks better, 2024.04.27 seems to have bugged link colors
+    "furo>=2023.0",  # Force it to be recent so that the theme looks better
     "tomli>=1.1",  # The load function doesn't work similarly before 1.1
-    "sphinx-autodoc-typehints>=1.16.0",  # Some problems with TypeVars before 1.16
+    "sphinx-autodoc-typehints>=3.5.0",  # Bugged Union on Python 3.14 before 3.5.0
     "myst-parser>=3.0.1",  # Never tested lower versions
     "sphinx-design>=0.6.0",  # Never tested lower versions
 ]