diff --git a/docs/conf.py b/docs/conf.py
index 559f3e15..87b73f2b 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -46,6 +46,7 @@ def setup(app):
                         "matplotlib.pyplot",
                         "tqdm",
                         "sympy",
+                        "sklean",
                         "sklearn.linear_model",
                         "torch.optim"]