easyblock = 'PythonPackage'

name = 'Keras'
version = '2.0.5'
versionsuffix = '-CNTK-1.2.0-Python-%(pyver)s'

homepage = 'https://keras.io/'
description = """Keras is a minimalist, highly modular neural networks library, written in Python and
capable of running on top of either TensorFlow or Theano."""

toolchain = {'name': 'foss', 'version': '2017a'}

source_urls = [PYPI_SOURCE]
sources = [SOURCE_TAR_GZ]

use_pip = True

dependencies = [
    ('Python', '3.6.1', '', True),
    ('CNTK', '2.0', '-Python-%(pyver)s'),
#   ('h5py', '2.7.0', '%(versionsuffix)s'), internal Python 3.6.1
#   ('PyYAML', '3.12', versionsuffix),      internal Python 3.6.1
]

# it defaults to Tensorflow
modextravars = {'KERAS_BACKEND': 'cntk'}

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

moduleclass = 'math'