add mockimport sklearn

KindXiaoming · Apr 29, 2024 · efcb597 · efcb597
1 parent 3a6ddea
commit efcb597
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
@@ -21,7 +21,7 @@ jobs:
       run: |
         pip install -U sphinx
         pip install sphinx-rtd-theme
-        pip install scikit-learn==1.0
+
     
     - name: Build docs
       run: |

diff --git a/docs/conf.py b/docs/conf.py
@@ -47,6 +47,7 @@ def setup(app):
                         "tqdm",
                         "sympy",
                         "scipy",
+                        "sklearn",
                         "torch.optim"]