diff --git a/tools/rapids-download-wheels-from-github b/tools/rapids-download-wheels-from-github index db81b36..82ed5f1 100755 --- a/tools/rapids-download-wheels-from-github +++ b/tools/rapids-download-wheels-from-github @@ -16,8 +16,11 @@ case "${pkg_type}" in cpp|python) pkg_name="$(rapids-package-name "wheel_${pkg_type}")" ;; + pure_python) + pkg_name="$(rapids-package-name "wheel_python" --pure)" + ;; *) - rapids-echo-stderr 'Pass "cpp" or "python" as an argument.' + rapids-echo-stderr 'Pass "cpp", "python", or "pure_python" as an argument.' exit 1 ;; esac