easyblock = 'PythonPackage'

name = 'Horovod'
version = '0.9.10'
versionsuffix = '-Python-%(pyver)s'

homepage = 'https://github.com/uber/horovod'
description = "Horovod is a distributed training framework for TensorFlow."

toolchain = {'name': 'intel', 'version': '2017b'}

source_urls = [PYPI_LOWER_SOURCE]
sources = [SOURCELOWER_TAR_GZ]
checksums = ['37c1e5efdb84fb33c770ca97eca883bd984ed91a84799653005a19ba366759b5']

dependencies = [
    ('Python', '3.6.3'),
    ('Tensorflow', '1.3.0', versionsuffix),
]

sanity_check_paths = {
    'files': [],
    'dirs': ['lib/python%(pyshortver)s/site-packages'],
}

moduleclass = 'tools'
