From c7bb94b86390f04634518abfac8d56c21bd0043b Mon Sep 17 00:00:00 2001
From: Nima Shoghi <nimashoghi@gmail.com>
Date: Wed, 27 Nov 2024 17:34:54 -0500
Subject: [PATCH] Update requirements.txt to include additional dependencies
 for documentation build

---
 docs/conf.py          | 6 +++---
 docs/requirements.txt | 5 +++++
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/docs/conf.py b/docs/conf.py
index 10f1e39..20d11c6 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -78,9 +78,9 @@ def setup(app):
 
 autodoc_mock_imports = [
     # "torch",
-    "pytorch_lightning",
-    "torch-geometric",
-    "torchmetrics",
+    # "pytorch_lightning",
+    # "torch-geometric",
+    # "torchmetrics",
 ]
 
 # Type hints settings
diff --git a/docs/requirements.txt b/docs/requirements.txt
index beab945..430df0e 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -4,5 +4,10 @@ myst-parser>=0.18.0
 sphinx-autodoc-typehints>=1.12.0
 sphinx-copybutton>=0.5.0
 nshconfig
+--index-url https://download.pytorch.org/whl/cpu
 torch
 ase
+scikit-learn
+lightning
+numpy
+torchmetrics