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"]