diff --git a/docs/requirements.txt b/docs/requirements.txt index 2b64d7e8f8ed..a9687eb9860c 100644 --- a/docs/requirements.txt +++ b/docs/requirements.txt @@ -12,5 +12,5 @@ pytest-xdist # Packages used for notebook execution matplotlib -sklearn +scikit-learn .[cpu] # Install jax from the current directory; jaxlib from pypi.