chore: debug jax[cpu]

abess-team · May 11, 2024 · fe786ec · fe786ec
1 parent 39cd3f9
commit fe786ec
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/source/userguide/install.rst b/docs/source/userguide/install.rst
@@ -73,4 +73,4 @@ The current minimum dependencies to run ``skscope`` are:
 
 - ``numpy``
 - ``scikit-learn>=1.2.2``
-- ``"jax"``
+- ``jax[cpu]``
diff --git a/setup.py b/setup.py
@@ -148,7 +148,7 @@ def build_extension(self, ext):
     install_requires=[
         "numpy",
         "scikit-learn>=1.2.2",
-        "jax",
+        "jax[cpu]",
         "scipy",
     ],
     license="MIT",