Update requirements.txt to include additional dependencies for docume…

…ntation build
Fung-Lab · Nov 27, 2024 · c7bb94b · c7bb94b
1 parent 1942d09
commit c7bb94b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git 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
@@ -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