pyrox.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.7";
16 name = "${pname}-${version}";
17
18 src = fetchPypi {
19 inherit pname version;
20 sha256 = "a6c72ee2b94be1ffefe7e77b69582b9827211f0c356b2189459711844d3634c0";
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}