diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh index f0afcd1159..23f3a13392 100755 --- a/ci/test_wheel.sh +++ b/ci/test_wheel.sh @@ -1,5 +1,5 @@ #!/bin/bash -# SPDX-FileCopyrightText: Copyright (c) 2023-2026, NVIDIA CORPORATION. +# SPDX-FileCopyrightText: Copyright (c) 2023-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved. # SPDX-License-Identifier: Apache-2.0 set -euo pipefail @@ -16,6 +16,17 @@ mkdir -p "${RAPIDS_TESTS_DIR}" # dependencies that can be installed later on when installing the wheel rapids-generate-pip-constraints test_python "${PIP_CONSTRAINT}" +python -m venv libcuml-env +. libcuml-env/bin/activate + +rapids-pip-retry install \ + -v \ + --prefer-binary \ + --constraint "${PIP_CONSTRAINT}" \ + "${LIBCUML_WHEELHOUSE}"/libcuml*.whl +python -c "import libcuml; assert (libraries := libcuml.load_library()) and all(libraries)" +deactivate + # notes: # # * just providing --constraint="${PIP_CONSTRAINT}" to be explicit, and because