diff --git a/ci/validate_wheel.sh b/ci/validate_wheel.sh index 708aeb029c..7f1dae43f2 100755 --- a/ci/validate_wheel.sh +++ b/ci/validate_wheel.sh @@ -10,30 +10,18 @@ package_name=$3 RAPIDS_CUDA_MAJOR="${RAPIDS_CUDA_VERSION%%.*}" # some packages are much larger on CUDA 11 than on CUDA 12 -# TODO(jameslamb): adjust these thresholds if [[ "${package_name}" == "libraft" ]]; then if [[ "${RAPIDS_CUDA_MAJOR}" == "11" ]]; then PYDISTCHECK_ARGS=( - --max-allowed-size-compressed '5.0G' + --max-allowed-size-compressed '750M' ) else PYDISTCHECK_ARGS=( - --max-allowed-size-compressed '5.0G' - ) - fi -elif [[ "${package_name}" == "pylibraft" ]] || [[ "${package_name}" == "raft-dask" ]]; then - if [[ "${RAPIDS_CUDA_MAJOR}" == "11" ]]; then - PYDISTCHECK_ARGS=( - --max-allowed-size-compressed '600M' - ) - else - PYDISTCHECK_ARGS=( - --max-allowed-size-compressed '400M' + --max-allowed-size-compressed '100M' ) fi else - echo "Unsupported package name: ${package_name}" - exit 1 + PYDISTCHECK_ARGS=() fi cd "${package_dir}" diff --git a/python/libraft/CMakeLists.txt b/python/libraft/CMakeLists.txt index 274638046e..bb9a87d36f 100644 --- a/python/libraft/CMakeLists.txt +++ b/python/libraft/CMakeLists.txt @@ -27,12 +27,10 @@ project( option(USE_CUDA_MATH_WHEELS "Use the CUDA math wheels instead of the system libraries" OFF) -# Check if raft is already available. If so, it is the user's responsibility to ensure that the -# CMake package is also available at build time of the Python raft package. - find_package(CUDAToolkit REQUIRED) -# TODO(jameslamb): make this work +# Check if raft is already available. If so, it is the user's responsibility to ensure that the +# CMake package is also available at build time of the Python raft package. find_package(raft "${RAPIDS_VERSION}") if(raft_FOUND) diff --git a/python/libraft/pyproject.toml b/python/libraft/pyproject.toml index 9479f0a832..e01692552c 100644 --- a/python/libraft/pyproject.toml +++ b/python/libraft/pyproject.toml @@ -123,6 +123,5 @@ matrix-entry = "cuda_suffixed=true;use_cuda_wheels=true" [tool.pydistcheck] select = [ # NOTE: size threshold is managed via CLI args in CI scripts - # TODO(jameslamb): adjust this "distro-too-large-compressed", ] diff --git a/python/pylibraft/pyproject.toml b/python/pylibraft/pyproject.toml index a3b1fa9372..912f1ad947 100644 --- a/python/pylibraft/pyproject.toml +++ b/python/pylibraft/pyproject.toml @@ -131,11 +131,12 @@ matrix-entry = "cuda_suffixed=true" [tool.pydistcheck] select = [ - # NOTE: size threshold is managed via CLI args in CI scripts - # TODO(jameslamb): adjust this "distro-too-large-compressed", ] +# PyPI limit is 100 MiB, fail CI before we get too close to that +max_allowed_size_compressed = '75M' + [tool.pytest.ini_options] filterwarnings = [ "error", diff --git a/python/raft-dask/pyproject.toml b/python/raft-dask/pyproject.toml index 8be374cce6..cc76c07d2b 100644 --- a/python/raft-dask/pyproject.toml +++ b/python/raft-dask/pyproject.toml @@ -30,7 +30,6 @@ authors = [ ] license = { text = "Apache 2.0" } requires-python = ">=3.10" -# TODO(jameslamb): does this really need joblib and numba? dependencies = [ "dask-cuda==25.2.*,>=0.0.0a0", "distributed-ucxx==0.42.*,>=0.0.0a0", @@ -136,5 +135,4 @@ select = [ ] # detect when package size grows significantly -# TODO(jameslamb): adjust this max_allowed_size_compressed = '300M'