diff options
Diffstat (limited to 'var')
| -rw-r--r-- | var/spack/repos/builtin/packages/py-dask-ml/package.py | 65 | ||||
| -rw-r--r-- | var/spack/repos/builtin/packages/py-dask-ml/xgboost_dependency.patch | 23 |
2 files changed, 88 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-dask-ml/package.py b/var/spack/repos/builtin/packages/py-dask-ml/package.py new file mode 100644 index 0000000000..6662eabe9c --- /dev/null +++ b/var/spack/repos/builtin/packages/py-dask-ml/package.py @@ -0,0 +1,65 @@ +# Copyright 2013-2020 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 PyDaskMl(PythonPackage): + """Scalable Machine Learning with Dask.""" + + homepage = "https://ml.dask.org/" + pypi = "dask-ml/dask-ml-1.8.0.tar.gz" + + version('1.8.0', sha256='8fc4ac3ec1915e382fb8cae9ff1ec9b5ac1bee0b6f4c6975d6e6cb7191a4a815') + + variant('docs', default=False, description='Build HTML documentation') + variant('xgboost', default=False, description='Deploys XGBoost alongside Dask') + + depends_on('python@3.6:', type=('build', 'run')) + + depends_on('py-setuptools', type='build') + depends_on('py-setuptools-scm', type='build') + + depends_on('py-dask+array+dataframe@2.4.0:', type=('build', 'run')) + depends_on('py-distributed@2.4.0:', type=('build', 'run')) + depends_on('py-numba', type=('build', 'run')) + depends_on('py-numpy@1.17.3:', type=('build', 'run')) + depends_on('py-pandas@0.24.2:', type=('build', 'run')) + depends_on('py-scikit-learn@0.23:', type=('build', 'run')) + depends_on('py-scipy', type=('build', 'run')) + depends_on('py-dask-glm@0.2.0:', type=('build', 'run')) + depends_on('py-multipledispatch@0.4.9:', type=('build', 'run')) + depends_on('py-packaging', type=('build', 'run')) + + depends_on('py-graphviz', type=('build', 'run'), when='+docs') + depends_on('py-heapdict', type=('build', 'run'), when='+docs') + depends_on('py-ipykernel', type=('build', 'run'), when='+docs') + depends_on('py-ipython', type=('build', 'run'), when='+docs') + depends_on('py-nbsphinx', type=('build', 'run'), when='+docs') + depends_on('py-nose', type=('build', 'run'), when='+docs') + depends_on('py-numpydoc', type=('build', 'run'), when='+docs') + depends_on('py-sortedcontainers', type=('build', 'run'), when='+docs') + depends_on('py-sphinx', type=('build', 'run'), when='+docs') + depends_on('py-sphinx-rtd-theme', type=('build', 'run'), when='+docs') + depends_on('py-sphinx-gallery', type=('build', 'run'), when='+docs') + depends_on('py-testpath', type=('build', 'run'), when='+docs') + depends_on('py-tornado', type=('build', 'run'), when='+docs') + depends_on('py-zict', type=('build', 'run'), when='+docs') + depends_on('py-dask-sphinx-theme@1.1.0:', type=('build', 'run'), when='+docs') + depends_on('py-nbsphinx', type=('build', 'run'), when='+docs') + + depends_on('py-xgboost+dask', type=('build', 'run'), when='+docs') + depends_on('py-xgboost+dask', type=('build', 'run'), when='+xgboost') + + patch('xgboost_dependency.patch') + + conflicts('+docs', when='%gcc target=aarch64:') + + @run_after('install') + def install_docs(self): + if '+docs' in self.spec: + with working_dir('docs'): + make('html') + install_tree('docs', self.prefix.docs) diff --git a/var/spack/repos/builtin/packages/py-dask-ml/xgboost_dependency.patch b/var/spack/repos/builtin/packages/py-dask-ml/xgboost_dependency.patch new file mode 100644 index 0000000000..39fa59c943 --- /dev/null +++ b/var/spack/repos/builtin/packages/py-dask-ml/xgboost_dependency.patch @@ -0,0 +1,23 @@ +diff --git a/dask_ml/xgboost.py b/dask_ml/xgboost.py +index 86e841db..70551df7 100644 +--- a/dask_ml/xgboost.py ++++ b/dask_ml/xgboost.py +@@ -4,4 +4,4 @@ This may be used for training an XGBoost model on a cluster. XGBoost + will be setup in distributed mode alongside your existing + ``dask.distributed`` cluster. + """ +-from dask_xgboost import * # noqa ++from xgboost import * # noqa +diff --git a/setup.py b/setup.py +index 857f6911..fb280973 100644 +--- a/setup.py ++++ b/setup.py +@@ -35,7 +35,7 @@ test_requires = [ + "pytest-mock", + ] + dev_requires = doc_requires + test_requires +-xgboost_requires = ["dask-xgboost", "xgboost"] ++xgboost_requires = ["xgboost[dask]"] + complete_requires = xgboost_requires + + extras_require = { |