Commit f9ca6e3d authored by rusty1s's avatar rusty1s
Browse files

[ci skip]

parent a44a5862
......@@ -11,7 +11,7 @@ if [ "${CUDA_VERSION}" = "cpu" ]; then
else
case $CUDA_VERSION in
cu115)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.3.*"
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.5.*"
;;
cu113)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.3.*"
......
......@@ -89,7 +89,7 @@ def get_extensions():
install_requires = []
test_require = [
test_requires = [
'pytest',
'pytest-cov',
]
......@@ -106,7 +106,7 @@ setup(
python_requires='>=3.7',
install_requires=install_requires,
extras_require={
'test': test_require,
'test': test_requires,
},
ext_modules=get_extensions() if not BUILD_DOCS else [],
cmdclass={
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment