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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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(
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
)
print("List of downloaded wheels:", filenames)
wheels.extend(wheels_partial)
print("List of downloaded wheels:", wheels)
print(
"""
Following steps should be done manually:

View File

@ -3,10 +3,11 @@ channels:
- conda-forge
dependencies:
- python=3.10
- pylint<3.2.4 # https://github.com/pylint-dev/pylint/issues/9751
- pylint
- wheel
- setuptools
- mypy
# https://github.com/python/mypy/issues/17617
- mypy !=1.11.0,!=1.11.1
- numpy
- scipy
- pandas