diff --git a/ci/build_wheel_nx-cugraph.sh b/ci/build_wheel_nx-cugraph.sh index 706adb793..55f9eb1ee 100755 --- a/ci/build_wheel_nx-cugraph.sh +++ b/ci/build_wheel_nx-cugraph.sh @@ -3,5 +3,5 @@ set -euo pipefail -# pkg-name pkg-dir ./ci/build_wheel.sh nx-cugraph . +./ci/validate_wheel.sh dist diff --git a/ci/validate_wheel.sh b/ci/validate_wheel.sh new file mode 100755 index 000000000..60a80fce6 --- /dev/null +++ b/ci/validate_wheel.sh @@ -0,0 +1,18 @@ +#!/bin/bash +# Copyright (c) 2024, NVIDIA CORPORATION. + +set -euo pipefail + +wheel_dir_relative_path=$1 + +rapids-logger "validate packages with 'pydistcheck'" + +pydistcheck \ + --inspect \ + "$(echo ${wheel_dir_relative_path}/*.whl)" + +rapids-logger "validate packages with 'twine'" + +twine check \ + --strict \ + "$(echo ${wheel_dir_relative_path}/*.whl)" diff --git a/pyproject.toml b/pyproject.toml index 91a211461..300e17a7a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -153,6 +153,14 @@ exclude_lines = [ "raise NotImplementedError", ] +[tool.pydistcheck] +select = [ + "distro-too-large-compressed", +] + +# PyPI limit is 100 MiB, fail CI before we get too close to that +max_allowed_size_compressed = '75M' + [tool.ruff] # https://github.com/charliermarsh/ruff/ line-length = 88