Update Documentation Building System (#78)

* Update mkdocs

* Move docs requirements to additional dependencies

* Add black dependency for API autoformat

* Change back to mknotebooks for latex in jupyter

Author: Ceyron

.github/workflows/build_docs.yml

@@ -24,8 +24,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install .
-          python -m pip install -r docs/requirements.txt
+          python -m pip install ".[docs]"
       
       - name: Build docs
         run: mkdocs build 

docs/requirements.txt

@@ -11,6 +11,17 @@ keywords = ["jax", "sciml", "deep-learning", "pde", "etdrk"]
 urls = {repository = "https://github.com/Ceyron/exponax" }
 dependencies = ["jax>=0.4.13", "jaxtyping>=0.2.20", "typing_extensions>=4.5.0", "equinox>=0.11.3", "matplotlib>=3.8.1"]
 
+[project.optional-dependencies]
+docs = [
+    "mkdocs==1.6.1",
+    "mkdocs-material==9.7.1",
+    "mkdocstrings==1.0.3",
+    "mkdocstrings-python==2.0.1",
+    "mknotebooks==0.8.0",
+    "griffe==1.15.0",
+    "black==26.1.0",  # Needed for API autoformat
+]
+
 [tool.ruff]
 line-length = 88
 target-version = "py310"