pyrox.dev
1{ lib, buildPythonPackage, fetchPypi
2, pytest, pytest-cov, pytest-xdist
3, six, numpy, scipy, pyyaml, h5py
4, keras-applications, keras-preprocessing
5}:
6
7buildPythonPackage rec {
8 pname = "keras";
9 version = "2.14.0";
10 format = "wheel";
11
12 src = fetchPypi {
13 inherit format pname version;
14 hash = "sha256-10KdHSExzH6x8uouwzAifH2dONqz398ueN7+5OzEP80=";
15 python = "py3";
16 dist = "py3";
17 };
18
19 nativeCheckInputs = [
20 pytest
21 pytest-cov
22 pytest-xdist
23 ];
24
25 propagatedBuildInputs = [
26 six pyyaml numpy scipy h5py
27 keras-applications keras-preprocessing
28 ];
29
30 # Couldn't get tests working
31 doCheck = false;
32
33 meta = with lib; {
34 description = "Deep Learning library for Theano and TensorFlow";
35 homepage = "https://keras.io";
36 license = licenses.mit;
37 maintainers = with maintainers; [ NikolaMandic ];
38 };
39}