Transition from requirements.txt to deps in pyproject.toml (#111)

Only deps in pyproject.toml are included when installing with `pip`.
This caused `clu` and other libraries to not be installed.

Fixes #110.

Author: jeffcarp

.github/workflows/ci.yml

@@ -18,7 +18,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install -r requirements.txt
+          pip install ".[dev]"
       - name: Run Unit Tests
         run: |
           pytest ./src/

.readthedocs.yaml

@@ -18,4 +18,7 @@ sphinx:
 # See https://docs.readthedocs.io/en/stable/guides/reproducible-builds.html
 python:
    install:
-   - requirements: docs/requirements.txt
+   - method: pip
+     path: .
+     extra_requirements:
+     - docs

README.md

@@ -27,6 +27,12 @@ pip install google-metrax
 
 ## Development
 
+Install the development dependencies:
+
+```sh
+pip install ".[dev]"
+```
+
 Run the tests:
 
 ```sh
@@ -36,7 +42,7 @@ pytest src/metrax
 Develop the docs locally:
 
 ```
-pip install -r ./docs/requirements.txt
+pip install ".[docs]"
 sphinx-build ./docs /tmp/metrax_docs
 python -m http.server --directory /tmp/metrax_docs
 ```

docs/requirements.txt

@@ -16,6 +16,12 @@ classifiers = [
     "Operating System :: OS Independent",
 ]
 license = {file = 'LICENSE'}
+dependencies = [
+  "clu==0.0.12",
+  "flax==0.11.1",
+  "jax==0.6.2",
+  "numpy==2.1.3",
+]
 
 [tool.hatch.build]
 include = [
@@ -32,6 +38,27 @@ packages = ["src/metrax"]
 Homepage = "https://github.com/google/metrax"
 Issues = "https://github.com/google/metrax/issues"
 
+[project.optional-dependencies]
+dev = [
+  "absl-py>=2.3.1",
+  "jax[cpu]==0.6.2",
+  "jax_tpu_embedding==0.1.0.dev20250618",
+  "keras-hub",
+  "keras-rs>=0.2.1",
+  "nltk>=3.9.1",
+  "pytest>=8.4.1",
+  "Pillow>=9.0.0",
+  "protobuf>=5.29.5",
+  "rouge-score>=0.1.2",
+  "scikit-learn>=1.7.1",
+  "tensorflow",
+  "torchmetrics>=1.8.1",
+]
+docs = [
+  "sphinx-rtd-theme==1.3.0rc1",
+  "scikit-learn==1.6.1",
+]
+
 [tool.ruff]
 indent-width = 2
 line-length = 120