Retag wheels to be cpXY

rapidsai · Nov 29, 2023 · 4bec5fc · 4bec5fc
1 parent 8f2a96e
commit 4bec5fc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -45,6 +45,10 @@ cd "${package_dir}"
 
 python -m pip wheel . -w dist -vvv --no-deps --disable-pip-version-check
 
+# Fix Python wheel tag to be restricted to CPython and version dependent
+WHEEL_PYTHON_TAG=`python -c 'import sys; print("cp{0}{1}".format(*sys.version_info[:2]))'`
+python -m wheel tags --remove --python-tag="${WHEEL_PYTHON_TAG}" --abi-tag="${WHEEL_PYTHON_TAG}" dist/*
+
 mkdir -p final_dist
 python -m auditwheel repair -w final_dist dist/*