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