# IT4Innovations 2018

easyblock = 'PythonPackage'

name = 'Keras'
version = '2.1.4'
versionsuffix = '-Tensorflow-1.6.0rc0-CUDA-9.0.176'

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': 'Py', 'version': '3.6'}

source_urls = [PYPI_SOURCE]
sources = [SOURCE_TAR_GZ]

dependencies = [
    ('Tensorflow', '1.6.0rc0', '-CUDA-9.0.176'),
    ('PyYAML', '3.12'),
    ('numpy', '1.13.3'),
    ('scipy', '1.0.0'),
    ('h5py', '2.7.1'),
]

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

sanity_check_paths = {
    'files': ['lib/python3.6/site-packages/%(name)s-%(version)s-py3.6.egg'],
    'dirs': [],
}

moduleclass = 'python'
