Shrink wheel size limit following removal of vector search APIs. (#2509)

Following #2498, we can apply this feedback from #2490: #2490 (comment)

These changes are inspired by rapidsai/cuvs#469.

Authors:
  - Bradley Dice (https://github.com/bdice)

Approvers:
  - Kyle Edwards (https://github.com/KyleFromNVIDIA)

URL: #2509

Author: bdice

ci/build_wheel_pylibraft.sh

@@ -18,4 +18,4 @@ esac
 export SKBUILD_CMAKE_ARGS="-DDETECT_CONDA_ENV=OFF;-DFIND_RAFT_CPP=OFF${EXTRA_CMAKE_ARGS}"
 
 ci/build_wheel.sh pylibraft ${package_dir}
-ci/validate_wheel.sh ${package_dir} final_dist
+ci/validate_wheel.sh ${package_dir} final_dist pylibraft

ci/build_wheel_raft_dask.sh

@@ -9,4 +9,4 @@ package_dir="python/raft-dask"
 export SKBUILD_CMAKE_ARGS="-DDETECT_CONDA_ENV=OFF;-DFIND_RAFT_CPP=OFF"
 
 ci/build_wheel.sh raft-dask ${package_dir}
-ci/validate_wheel.sh ${package_dir} final_dist
+ci/validate_wheel.sh ${package_dir} final_dist raft-dask

ci/validate_wheel.sh

@@ -5,13 +5,37 @@ set -euo pipefail
 
 package_dir=$1
 wheel_dir_relative_path=$2
+package_name=$3
+
+RAPIDS_CUDA_MAJOR="${RAPIDS_CUDA_VERSION%%.*}"
+
+# some packages are much larger on CUDA 11 than on CUDA 12
+if [[ "${package_name}" == "raft-dask" ]]; then
+    PYDISTCHECK_ARGS=(
+        --max-allowed-size-compressed '200M'
+    )
+elif [[ "${package_name}" == "pylibraft" ]]; then
+    if [[ "${RAPIDS_CUDA_MAJOR}" == "11" ]]; then
+        PYDISTCHECK_ARGS=(
+            --max-allowed-size-compressed '600M'
+        )
+    else
+        PYDISTCHECK_ARGS=(
+            --max-allowed-size-compressed '100M'
+        )
+    fi
+else
+    echo "Unsupported package name: ${package_name}"
+    exit 1
+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'"

python/pylibraft/pyproject.toml

@@ -132,12 +132,10 @@ 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",
 ]
 
-# detect when package size grows significantly
-max_allowed_size_compressed = '825M'
-
 [tool.pytest.ini_options]
 filterwarnings = [
     "error",