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