refactor: switch to optional dependencies

Signed-off-by: nstarman <[email protected]>

Author: nstarman

.github/workflows/release.yml

@@ -16,7 +16,7 @@ jobs:
             test-script: |
                 cp -r ${{ github.workspace }}/test ./test
                 cp ${{ github.workspace }}/pyproject.toml ./pyproject.toml
-                python -m pip install -r ./test/requirements.txt
+                python -m pip install '.[tests]'
                 python -m test
             pypi-token: ${{ secrets.pypi_token }}
             github-user: patrick-kidger

.github/workflows/run_tests.yml

@@ -23,8 +23,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install -r ./test/requirements.txt
-
+          python -m pip install '.[tests]'
 
       - name: Checks with pre-commit
         uses: pre-commit/[email protected]

CONTRIBUTING.md

@@ -34,7 +34,7 @@ Now make your changes. Make sure to include additional tests if necessary.
 Next verify the tests all pass:
 
 ```bash
-pip install -r test/requirements.txt
+pip install -e '.[tests]'
 pytest
 ```
 

pyproject.toml

@@ -40,6 +40,14 @@ docs = [
   "mkdocstrings-python==1.16.8",
   "pymdown-extensions==10.14.3"
 ]
+tests = [
+    "beartype>=0.22.5",
+    "jaxlib>=0.6.2",
+    "optax>=0.2.6",
+    "pytest>=9.0.1",
+    "scipy>=1.15.3",
+    "tqdm>=4.67.1",
+]
 
 [tool.hatch.build]
 include = ["diffrax/*"]