Update to latest release workflow + bump to Lineax compatible with latest JAX

Author: patrick-kidger

Diff for: .github/workflows/release.yml

@@ -10,7 +10,7 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Release
-        uses: patrick-kidger/action_update_python_project@v2
+        uses: patrick-kidger/action_update_python_project@v4
         with:
             python-version: "3.11"
             test-script: |
@@ -21,7 +21,3 @@ jobs:
             pypi-token: ${{ secrets.pypi_token }}
             github-user: patrick-kidger
             github-token: ${{ github.token }}
-            email-user: ${{ secrets.email_user }}
-            email-token: ${{ secrets.email_token }}
-            email-server: ${{ secrets.email_server }}
-            email-target: ${{ secrets.email_target }}

Diff for: .pre-commit-config.yaml

@@ -16,11 +16,11 @@ repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
     rev: v0.1.7
     hooks:
-      - id: ruff  # linter
+      - id: ruff-format
         types_or: [ python, pyi, jupyter ]
-        args: [ --fix ]
-      - id: ruff-format  # formatter
+      - id: ruff
         types_or: [ python, pyi, jupyter ]
+        args: [ --fix ]
   - repo: https://github.com/RobertCraigie/pyright-python
     rev: v1.1.331
     hooks:

Diff for: pyproject.toml

@@ -33,7 +33,7 @@ classifiers = [
     "Topic :: Scientific/Engineering :: Mathematics",
 ]
 urls = {repository = "https://github.com/patrick-kidger/optimistix" }
-dependencies = ["jax>=0.4.28", "jaxtyping>=0.2.23", "lineax>=0.0.5", "equinox>=0.11.7", "typing_extensions>=4.5.0"]
+dependencies = ["jax>=0.4.28", "jaxtyping>=0.2.23", "lineax>=0.0.6", "equinox>=0.11.7", "typing_extensions>=4.5.0"]
 
 [build-system]
 requires = ["hatchling"]

Diff for: tests/test_solve.py

@@ -56,10 +56,9 @@ def test_forward_mode():
     def f(y, _):
         return eqxi.nondifferentiable_backward(y)
 
-    sol = optx.least_squares(
+    optx.least_squares(
         f,
         optx.LevenbergMarquardt(rtol=1e-4, atol=1e-4),
         jnp.arange(3.0),
         options=dict(jac="fwd"),
     )
-    return sol.value