deps: migrate to tests group (#345)

* deps: migrate to tests group

Signed-off-by: nstarman <nstarman@users.noreply.github.com>

* config: change to optional dependencies

Signed-off-by: nstarman <nstarman@users.noreply.github.com>

* config(tests): use mlx[cpu]

See ml-explore/mlx#2402

Signed-off-by: nstarman <nstarman@users.noreply.github.com>

---------

Signed-off-by: nstarman <nstarman@users.noreply.github.com>

Author: nstarman

.github/workflows/release.yml

@@ -33,7 +33,7 @@ jobs:
         with:
             python-version: "3.11"
             test-script: |
-                python -m pip install -r ${{ github.workspace }}/test/requirements.txt
+                python -m pip install .[tests]
                 python -m pip install torch --extra-index-url https://download.pytorch.org/whl/cpu
                 cp -r ${{ github.workspace }}/test ./test
                 pytest

.github/workflows/run_tests.yml

@@ -42,7 +42,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]
           python -m pip install torch --extra-index-url https://download.pytorch.org/whl/cpu
 
       - name: Checks with pre-commit

CONTRIBUTING.md

@@ -28,7 +28,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]
 pip install torch --extra-index-url https://download.pytorch.org/whl/cpu
 pytest
 ```

pyproject.toml

@@ -27,6 +27,9 @@ dependencies = ["wadler_lindig>=0.1.3"]
 entry-points = {pytest11 = {jaxtyping = "jaxtyping._pytest_plugin"}}
 
 [project.optional-dependencies]
+dev = [
+  "pre-commit>=4.3.0",
+]
 docs = [
   "hippogriffe==0.2.0",
   "mkdocs==1.6.1",
@@ -36,6 +39,20 @@ docs = [
   "mkdocstrings[python]==0.28.3",
   "pymdown-extensions==10.14.3",
 ]
+tests = [
+    "beartype>=0.21.0",
+    "cloudpickle>=3.1.1",
+    "equinox>=0.13.1",
+    "ipython>=8.37.0",
+    "jax>=0.5.3",
+    "mlx[cpu]>=0.29.1",
+    "numpy<2",
+    "pytest>=8.4.2",
+    "pytest-asyncio>=1.2.0",
+    "tensorflow>=2.18.1",
+    "typeguard<3",
+]
+
 
 [build-system]
 requires = ["hatchling"]