tjh.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.11.0";
10 format = "wheel";
11
12 src = fetchPypi {
13 inherit format pname version;
14 hash = "sha256-OMb/8OqaiwaicXc2VlySpzyM2bHCOecSXMsYi3hI9l4=";
15 };
16
17 nativeCheckInputs = [
18 pytest
19 pytest-cov
20 pytest-xdist
21 ];
22
23 propagatedBuildInputs = [
24 six pyyaml numpy scipy h5py
25 keras-applications keras-preprocessing
26 ];
27
28 # Couldn't get tests working
29 doCheck = false;
30
31 meta = with lib; {
32 description = "Deep Learning library for Theano and TensorFlow";
33 homepage = "https://keras.io";
34 license = licenses.mit;
35 maintainers = with maintainers; [ NikolaMandic ];
36 };
37}