Update the release script to download xgboost-cpu (#10657)

* Update the release script to download xgboost-cpu

* Exclude mypy 1.11.1; un-cap pylint

* Exclude mypy 1.11.0 too
This commit is contained in:
Philip Hyunsu Cho
2024-07-31 14:43:10 -07:00
committed by GitHub
parent fb77ed7603
commit 778751a1bb
2 changed files with 19 additions and 12 deletions

View File

@@ -135,7 +135,8 @@ def make_pysrc_wheel(
def download_py_packages(
branch: str, major: int, minor: int, commit_hash: str, outdir: str
) -> None:
platforms = [
# List of platforms for full package and minimal package
full_platforms = [
"win_amd64",
"manylinux2014_x86_64",
"manylinux2014_aarch64",
@@ -144,19 +145,24 @@ def download_py_packages(
"macosx_10_15_x86_64.macosx_11_0_x86_64.macosx_12_0_x86_64",
"macosx_12_0_arm64",
]
minimal_platforms = [
"win_amd64",
"manylinux2014_x86_64",
"manylinux2014_aarch64",
]
branch = branch.split("_")[1] # release_x.y.z
dir_URL = PREFIX + branch + "/"
src_filename_prefix = "xgboost-" + args.release + "%2B" + commit_hash + "-py3-none-"
target_filename_prefix = "xgboost-" + args.release + "-py3-none-"
wheels = []
if not os.path.exists(DIST):
os.mkdir(DIST)
filenames = download_wheels(
platforms, dir_URL, src_filename_prefix, target_filename_prefix, outdir
)
print("List of downloaded wheels:", filenames)
for pkg_name, platforms in [("xgboost", full_platforms), ("xgboost_cpu", minimal_platforms)]:
src_filename_prefix = f"{pkg_name}-{args.release}%2B{commit_hash}-py3-none-"
target_filename_prefix = f"{pkg_name}-{args.release}-py3-none-"
wheels_partial = download_wheels(
platforms, dir_URL, src_filename_prefix, target_filename_prefix, outdir
)
wheels.extend(wheels_partial)
print("List of downloaded wheels:", wheels)
print(
"""
Following steps should be done manually: