Cleanup PyPI variant script. (#11294)

Author: trivialfis

ops/script/pypi_variants.py

@@ -3,19 +3,20 @@
 import argparse
 import os
 
-from test_utils import PY_PACKAGE, ROOT
+from test_utils import PY_PACKAGE
 
 IN_PATH = os.path.join(PY_PACKAGE, "pyproject.toml.in")
 OUT_PATH = os.path.join(PY_PACKAGE, "pyproject.toml")
 
-WHL_CPU = """
-[tool.hatch.build.targets.wheel]
-packages = ["xgboost/"]
-"""
-
 CHOICES = ["default", "cpu", "manylinux2014"]
 
 
+NCCL_WHL = """    \"nvidia-nccl-cu12 ; platform_system == 'Linux' and platform_machine != 'aarch64'\","""
+
+NAME = "{{ name }}"
+NCCL = "{{ nccl }}"
+
+
 def copyfile(src: str, dst: str) -> None:
     with open(src, "rb") as fd:
         content = fd.read()
@@ -33,28 +34,13 @@ def make_pyproject(variant: str) -> None:
     readme_cpu = os.path.join(PY_PACKAGE, "README.cpu.rst")
     readme = os.path.join(PY_PACKAGE, "README.rst")
     if variant == "cpu":
-        pyproject = (
-            pyproject.replace("{{ name }}", "xgboost-cpu")
-            .replace("{{ wheel }}", WHL_CPU)
-            .replace("{{ nccl }}", "")
-        )
+        pyproject = pyproject.replace(NAME, "xgboost-cpu").replace(NCCL, "")
         copyfile(readme_cpu, readme)
     elif variant == "manylinux2014":
-        pyproject = (
-            pyproject.replace("{{ name }}", "xgboost")
-            .replace("{{ wheel }}", "")
-            .replace("{{ nccl }}", "")
-        )
+        pyproject = pyproject.replace(NAME, "xgboost").replace(NCCL, "")
         copyfile(readme_dft, readme)
     else:
-        pyproject = (
-            pyproject.replace("{{ name }}", "xgboost")
-            .replace("{{ wheel }}", "")
-            .replace(
-                "{{ nccl }}",
-                """    \"nvidia-nccl-cu12 ; platform_system == 'Linux' and platform_machine != 'aarch64'\",""",
-            )
-        )
+        pyproject = pyproject.replace(NAME, "xgboost").replace(NCCL, NCCL_WHL)
         copyfile(readme_dft, readme)
     pyproject = (
         f"# Generated by `{os.path.basename(__file__)}`, don't edit.\n" + pyproject

python-package/pyproject.toml

@@ -73,7 +73,8 @@ disable = [
 dummy-variables-rgx = "(unused|)_.*"
 reports = false
 
-
+[tool.hatch.build.targets.wheel]
+packages = ["xgboost/"]
 
 [tool.pylint.basic]
 # Enforce naming convention

python-package/pyproject.toml.in

@@ -72,7 +72,8 @@ disable = [
 dummy-variables-rgx = "(unused|)_.*"
 reports = false
 
-{{ wheel }}
+[tool.hatch.build.targets.wheel]
+packages = ["xgboost/"]
 
 [tool.pylint.basic]
 # Enforce naming convention