diff --git a/ci/validate_wheel.sh b/ci/validate_wheel.sh index 914abe3c1c..ec3867aa30 100755 --- a/ci/validate_wheel.sh +++ b/ci/validate_wheel.sh @@ -9,28 +9,12 @@ package_name=$3 RAPIDS_CUDA_MAJOR="${RAPIDS_CUDA_VERSION%%.*}" -# some packages are much larger on CUDA 11 than on CUDA 12 -PYDISTCHECK_ARGS=() -if [[ "${package_name}" == "libraft" ]]; then - # TODO(jameslamb): revise these thresholds - if [[ "${RAPIDS_CUDA_MAJOR}" == "11" ]]; then - PYDISTCHECK_ARGS+=( - --max-allowed-size-compressed '750M' - ) - else - PYDISTCHECK_ARGS+=( - --max-allowed-size-compressed '100M' - ) - fi -fi - cd "${package_dir}" rapids-logger "validate packages with 'pydistcheck'" pydistcheck \ --inspect \ - "${PYDISTCHECK_ARGS[@]}" \ "$(echo ${wheel_dir_relative_path}/*.whl)" rapids-logger "validate packages with 'twine'" diff --git a/python/libraft/pyproject.toml b/python/libraft/pyproject.toml index 549a1bf651..89b2834614 100644 --- a/python/libraft/pyproject.toml +++ b/python/libraft/pyproject.toml @@ -110,6 +110,8 @@ matrix-entry = "cuda_suffixed=true;use_cuda_wheels=true" [tool.pydistcheck] select = [ - # NOTE: size threshold is managed via CLI args in CI scripts "distro-too-large-compressed", ] + +# PyPI limit is 100 MiB, fail CI before we get too close to that +max_allowed_size_compressed = '75M'