Martin Weinelt
56 lines
903 B
Nix
56 lines
903 B
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchPypi,
|
|
pytest,
|
|
pytest-cov,
|
|
pytest-xdist,
|
|
six,
|
|
numpy,
|
|
scipy,
|
|
pyyaml,
|
|
h5py,
|
|
optree,
|
|
keras-applications,
|
|
keras-preprocessing,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "keras";
|
|
version = "3.2.1";
|
|
format = "wheel";
|
|
|
|
src = fetchPypi {
|
|
inherit format pname version;
|
|
hash = "sha256-C+HomwQeaXvlYthCLsuVjuVIGs/AiZEyAJJsVh0ligM=";
|
|
python = "py3";
|
|
dist = "py3";
|
|
};
|
|
|
|
nativeCheckInputs = [
|
|
pytest
|
|
pytest-cov
|
|
pytest-xdist
|
|
];
|
|
|
|
propagatedBuildInputs = [
|
|
six
|
|
pyyaml
|
|
numpy
|
|
scipy
|
|
h5py
|
|
keras-applications
|
|
keras-preprocessing
|
|
];
|
|
|
|
# Couldn't get tests working
|
|
doCheck = false;
|
|
|
|
meta = with lib; {
|
|
description = "Deep Learning library for Theano and TensorFlow";
|
|
homepage = "https://keras.io";
|
|
license = licenses.mit;
|
|
maintainers = with maintainers; [ NikolaMandic ];
|
|
};
|
|
}
|