diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
index 55ec532f..516d5282 100644
--- a/.github/workflows/docs.yml
+++ b/.github/workflows/docs.yml
@@ -21,6 +21,7 @@ jobs:
       run: |
         pip3 install -U sphinx
         pip3 install sphinx-rtd-theme
+        pip3 install scikit-learn
     
     - name: Build docs
       run: |
diff --git a/docs/conf.py b/docs/conf.py
index 47a54a66..cde5db4c 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -46,8 +46,6 @@ def setup(app):
                         "matplotlib.pyplot",
                         "tqdm",
                         "sympy",
-                        "scikit-learn",
-                        "sklearn.linear_model",
                         "torch.optim"]