# Copyright 2013-2022 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)