hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

New Package: py-xgboost (#21470)

Browse source 
* new python package py-xgboost added

* py-xgboost spack style applied

* py-xgboost: Addressing PR #21470 comments. Added variant descriptions and fixed scikit-learn when clause
This commit is contained in:
kean-smullen-nnl 2021-02-09 18:12:20 -05:00 committed by GitHub
parent 42e8a64660
commit 32056da93b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 39 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

39
var/spack/repos/builtin/packages/py-xgboost/package.py Normal file
View file

@ -0,0 +1,39 @@
# Copyright 2013-2021 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)
from spack 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'
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.')
depends_on('cmake', type='build')
depends_on('python@3.6:', type=('build', 'run'))
depends_on('py-setuptools', 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'))