easyblock = 'PythonPackage'

name = 'Keras'
version = '2.1.2'
versionsuffix = '-Tensorflow-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 = ['https://github.com/keras-team/keras/archive']
sources = ['%(version)s.tar.gz']

use_pip = True

dependencies = [
    ('Python', '3.6.1'),
    ('Tensorflow', '1.2.0', '', ('GCC', '7.1.0-2.28')),
    #   ('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': 'tensorflow'}

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

moduleclass = 'math'