tjh.dev
1{ stdenv
2, buildPythonPackage
3, fetchPypi
4, pytest
5, pytestcov
6, pytestpep8
7, pytest_xdist
8, six
9, Theano
10, pyyaml
11}:
12
13buildPythonPackage rec {
14 pname = "Keras";
15 version = "2.0.8";
16 name = "${pname}-${version}";
17
18 src = fetchPypi {
19 inherit pname version;
20 sha256 = "899dc6aaed366f20100b9f80cf1093ea5b43eecc74afd1dc63a4e48dfa776ab9";
21 };
22
23 checkInputs = [
24 pytest
25 pytestcov
26 pytestpep8
27 pytest_xdist
28 ];
29
30 propagatedBuildInputs = [
31 six Theano pyyaml
32 ];
33
34 # Couldn't get tests working
35 doCheck = false;
36
37 meta = with stdenv.lib; {
38 description = "Deep Learning library for Theano and TensorFlow";
39 homepage = https://keras.io;
40 license = licenses.mit;
41 maintainers = with maintainers; [ NikolaMandic ];
42 };
43}