diff options
Diffstat (limited to 'nixpkgs/pkgs/development/python-modules/keras/default.nix')
-rw-r--r-- | nixpkgs/pkgs/development/python-modules/keras/default.nix | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/nixpkgs/pkgs/development/python-modules/keras/default.nix b/nixpkgs/pkgs/development/python-modules/keras/default.nix new file mode 100644 index 000000000000..dde820876d81 --- /dev/null +++ b/nixpkgs/pkgs/development/python-modules/keras/default.nix @@ -0,0 +1,37 @@ +{ stdenv, buildPythonPackage, fetchPypi +, pytest, pytestcov, pytestpep8, pytest_xdist +, six, numpy, scipy, pyyaml, h5py +, keras-applications, keras-preprocessing +}: + +buildPythonPackage rec { + pname = "Keras"; + version = "2.2.4"; + + src = fetchPypi { + inherit pname version; + sha256 = "90b610a3dbbf6d257b20a079eba3fdf2eed2158f64066a7c6f7227023fd60bc9"; + }; + + checkInputs = [ + pytest + pytestcov + pytestpep8 + pytest_xdist + ]; + + propagatedBuildInputs = [ + six pyyaml numpy scipy h5py + keras-applications keras-preprocessing + ]; + + # Couldn't get tests working + doCheck = false; + + meta = with stdenv.lib; { + description = "Deep Learning library for Theano and TensorFlow"; + homepage = https://keras.io; + license = licenses.mit; + maintainers = with maintainers; [ NikolaMandic ]; + }; +} |