setuptools only needed for cuda really

ratt-ru · Oct 29, 2024 · 5781613 · 5781613
1 parent c843a25
commit 5781613
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -16,7 +16,7 @@
 ]
 
 extras_require = {
-    "cuda": ["cupy >= 9.0.0", "jinja2 >= 2.10"],
+    "cuda": ["cupy >= 9.0.0", "jinja2 >= 2.10", "setuptools"],
     "dask": ["dask[array] >= 2.2.0"],
     "jax": ["jax >= 0.2.11", "jaxlib >= 0.1.65"],
     "scipy": ["scipy >= 1.4.0"],