Ensure that pip check does not fail due to bad platform tag (dmlc#10755) (dmlc#10902)

---------

Co-authored-by: Philip Hyunsu Cho <[email protected]>
Co-authored-by: jakirkham <[email protected]>

Author: 3 people

python-package/hatch_build.py

@@ -3,16 +3,16 @@
 Here, we customize the tag of the generated wheels.
 """
 
-import sysconfig
 from typing import Any, Dict
 
 from hatchling.builders.hooks.plugin.interface import BuildHookInterface
+from packaging.tags import platform_tags
 
 
 def get_tag() -> str:
     """Get appropriate wheel tag according to system"""
-    tag_platform = sysconfig.get_platform().replace("-", "_").replace(".", "_")
-    return f"py3-none-{tag_platform}"
+    platform_tag = next(platform_tags())
+    return f"py3-none-{platform_tag}"
 
 
 class CustomBuildHook(BuildHookInterface):

python-package/pyproject.toml

@@ -1,6 +1,7 @@
 [build-system]
 requires = [
-    "hatchling>=1.12.1"
+    "hatchling>=1.12.1",
+    "packaging>=21.3",
 ]
 backend-path = ["."]
 build-backend = "packager.pep517"
@@ -30,7 +31,7 @@ classifiers = [
 dependencies = [
     "numpy",
     "scipy",
-    "nvidia-nccl-cu12 ; platform_system == 'Linux' and platform_machine != 'aarch64'"
+    "nvidia-nccl-cu12 ; platform_system == 'Linux' and platform_machine != 'aarch64'",
 ]
 
 [project.urls]

tests/buildkite/remove_nccl_dep.patch

@@ -1,14 +1,12 @@
 diff --git python-package/pyproject.toml python-package/pyproject.toml
-index 8835def25..46c1451c2 100644
+index 20d3f9974..953087ff4 100644
 --- python-package/pyproject.toml
 +++ python-package/pyproject.toml
-@@ -30,8 +30,7 @@ classifiers = [
- ]
+@@ -30,7 +30,6 @@ classifiers = [
  dependencies = [
      "numpy",
--    "scipy",
--    "nvidia-nccl-cu12 ; platform_system == 'Linux' and platform_machine != 'aarch64'"
-+    "scipy"
+     "scipy",
+-    "nvidia-nccl-cu12 ; platform_system == 'Linux' and platform_machine != 'aarch64'",
  ]
 
  [project.urls]