Add cupy to Python test dependencies.

rapidsai · Dec 13, 2022 · 8032473 · 8032473
1 parent 6899d56
commit 8032473
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/conda/environments/all_cuda-115_arch-x86_64.yaml b/conda/environments/all_cuda-115_arch-x86_64.yaml
@@ -14,6 +14,7 @@ dependencies:
 - cmake>=3.23.1,!=3.25.0
 - cuda-python >=11.7.1,<12.0
 - cudatoolkit=11.5
+- cupy
 - cxx-compiler
 - cython>=0.29,<0.30
 - dask-cuda=23.02.*

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -150,6 +150,7 @@ dependencies:
     common:
       - output_types: [conda, requirements]
         packages:
+          - cupy
           - pytest
           - pytest-cov
           - scikit-learn