From 80324737e423139e6d805a91b6b760dbc3f90fd1 Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Tue, 13 Dec 2022 11:15:43 -0600 Subject: [PATCH] Add cupy to Python test dependencies. --- conda/environments/all_cuda-115_arch-x86_64.yaml | 1 + dependencies.yaml | 1 + 2 files changed, 2 insertions(+) diff --git a/conda/environments/all_cuda-115_arch-x86_64.yaml b/conda/environments/all_cuda-115_arch-x86_64.yaml index eafb3572b3..7cf230e2c5 100644 --- 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 index ccf6796c49..a59d634250 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -150,6 +150,7 @@ dependencies: common: - output_types: [conda, requirements] packages: + - cupy - pytest - pytest-cov - scikit-learn