diff options
Diffstat (limited to 'infra/libkookie/nixpkgs/pkgs/development/python-modules/keras/default.nix')
-rw-r--r-- | infra/libkookie/nixpkgs/pkgs/development/python-modules/keras/default.nix | 37 |
1 files changed, 37 insertions, 0 deletions
diff --git a/infra/libkookie/nixpkgs/pkgs/development/python-modules/keras/default.nix b/infra/libkookie/nixpkgs/pkgs/development/python-modules/keras/default.nix new file mode 100644 index 000000000000..8b6301ab59b7 --- /dev/null +++ b/infra/libkookie/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.4.3"; + + src = fetchPypi { + inherit pname version; + sha256 = "fedd729b52572fb108a98e3d97e1bac10a81d3917d2103cc20ab2a5f03beb973"; + }; + + 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 ]; + }; +} |