tjh.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.4.3";
10
11 src = fetchPypi {
12 inherit pname version;
13 sha256 = "fedd729b52572fb108a98e3d97e1bac10a81d3917d2103cc20ab2a5f03beb973";
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}