diff --git a/.github/workflows/docs.yaml b/.github/workflows/docs.yaml
index 49a43b7..70123d0 100644
--- a/.github/workflows/docs.yaml
+++ b/.github/workflows/docs.yaml
@@ -25,12 +25,13 @@ jobs:
 
     - name: Install dependencies
       run: |
-        pip install jupyter-book
+        pip install poetry
+        poetry install --only doc
 
     # Build the book
     - name: Build the book
       run: |
-        jupyter-book build docs
+        poetry run jupyter-book build docs
 
     # Upload the book's HTML as an artifact
     - name: Upload artifact
diff --git a/poetry.lock b/poetry.lock
index d2e7011..2a7ad49 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -730,20 +730,6 @@ docs = ["furo (>=2024.8.6)", "sphinx (>=8.0.2)", "sphinx-autodoc-typehints (>=2.
 testing = ["covdefaults (>=2.3)", "coverage (>=7.6.1)", "diff-cover (>=9.2)", "pytest (>=8.3.3)", "pytest-asyncio (>=0.24)", "pytest-cov (>=5)", "pytest-mock (>=3.14)", "pytest-timeout (>=2.3.1)", "virtualenv (>=20.26.4)"]
 typing = ["typing-extensions (>=4.12.2)"]
 
-[[package]]
-name = "fire"
-version = "0.6.0"
-description = "A library for automatically generating command line interfaces."
-optional = false
-python-versions = "*"
-files = [
-    {file = "fire-0.6.0.tar.gz", hash = "sha256:54ec5b996ecdd3c0309c800324a0703d6da512241bc73b553db959d98de0aa66"},
-]
-
-[package.dependencies]
-six = "*"
-termcolor = "*"
-
 [[package]]
 name = "fonttools"
 version = "4.54.1"
@@ -4774,4 +4760,4 @@ type = ["pytest-mypy"]
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.10"
-content-hash = "aa77c3311618b388d51c1739124cae9be0406e50e003b8ee2ee6eb1915aa7a90"
+content-hash = "12935bf20e5ac4351734a47b1a71486daccbc0eed1239abe67858b3b17df8b37"
diff --git a/pyproject.toml b/pyproject.toml
index 0597021..ef55b3c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,7 +16,6 @@ script = "build.py"
 
 [tool.poetry.group.dev.dependencies]
 # Experiment related dependencies here to not clutter the main project dependencies.
-fire = "0.6.0"
 notebook = "^7.2.1"
 numpy = "^1.26.4"
 openai = "^1.31.0"
@@ -30,6 +29,10 @@ termcolor = "^2.4.0"
 matplotlib = "^3.9.2"
 seaborn = "^0.13.2"
 pandas = "^2.2.3"
+
+[tool.poetry.group.doc]
+optional = true
+[tool.poetry.group.doc.dependencies]
 jupyter-book = "^1.0.3"
 
 [build-system]