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