add mockimport sklearn

KindXiaoming · Apr 29, 2024 · 5ce7d7d · 5ce7d7d
1 parent 490e10c
commit 5ce7d7d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -46,6 +46,8 @@ def setup(app):
                         "matplotlib.pyplot",
                         "tqdm",
                         "sympy",
+                        "sklean",
+                        "sklearn.linear_model"
                         "torch.optim"]