update thresholds

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}"

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)

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",
 ]

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",

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'