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

View File

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