easyblock = 'PythonPackage'

name = 'Keras'
version = '2.0.5'
versionsuffix = '-Theano-1.2.0-Python-%(local_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 = SYSTEM

source_urls = [PYPI_SOURCE]
sources = [SOURCE_TAR_GZ]

use_pip = True

dependencies = [
    ('Python', '3.6.1'),
    ('Theano', '0.9.0', '-Python-%(local_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': 'theano'}

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

moduleclass = 'math'