1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
import os
from spack.package import *
class PyXgboost(PythonPackage):
"""XGBoost is an optimized distributed gradient boosting library designed to be
highly efficient, flexible and portable."""
homepage = "https://xgboost.ai/"
pypi = "xgboost/xgboost-1.3.3.tar.gz"
maintainers("adamjstewart")
import_modules = ["xgboost"]
license("Apache-2.0")
version("1.6.2", sha256="e1f5c91ba88cf8edb409d7fd2ca150dcd80b6f2115587d87365f0c10b2d4f009")
version("1.6.1", sha256="24072028656f3428e7b8aabf77340ece057f273e41f7f85d67ccaefb7454bb18")
version("1.5.2", sha256="404dc09dca887ef5a9bc0268f882c54b33bfc16ac365a859a11e7b24d49da387")
version("1.3.3", sha256="397051647bb837915f3ff24afc7d49f7fca57630ffd00fb5ef66ae2a0881fb43")
variant("pandas", default=False, description="Enable Pandas extensions for training.")
variant(
"scikit-learn", default=False, description="Enable scikit-learn extensions for training."
)
variant("dask", default=False, description="Enables Dask extensions for distributed training.")
variant("plotting", default=False, description="Enables tree and importance plotting.")
for ver in ["1.3.3", "1.5.2", "1.6.1", "1.6.2"]:
depends_on("xgboost@" + ver, when="@" + ver)
depends_on("python@3.7:", when="@1.6:", type=("build", "run"))
depends_on("python@3.6:", type=("build", "run"))
depends_on("py-setuptools", type=("build"))
# in newer pip versions --install-option does not exist
depends_on("py-pip@:23.0", type="build")
depends_on("py-numpy", type=("build", "run"))
depends_on("py-scipy", type=("build", "run"))
depends_on("py-pandas", when="+pandas", type=("build", "run"))
depends_on("py-scikit-learn", when="+scikit-learn", type=("build", "run"))
depends_on("py-dask", when="+dask", type=("build", "run"))
depends_on("py-pandas", when="+dask", type=("build", "run"))
depends_on("py-distributed", when="+dask", type=("build", "run"))
depends_on("py-graphviz", when="+plotting", type=("build", "run"))
depends_on("py-matplotlib", when="+plotting", type=("build", "run"))
def patch(self):
# https://github.com/dmlc/xgboost/issues/6706
# 'setup.py' is hard-coded to search in Python installation prefix
filter_file(
"lib_path = os.path.join(sys.prefix, 'lib')",
"lib_path = '{0}'".format(self.spec["xgboost"].libs.directories[0]),
"setup.py",
string=True,
)
# Same for run-time search
filter_file(
"os.path.join(curr_path, 'lib'),",
"'{0}',".format(self.spec["xgboost"].libs.directories[0]),
os.path.join("xgboost", "libpath.py"),
string=True,
)
def install_options(self, spec, prefix):
return ["--use-system-libxgboost"]
|
