tjh.dev
1{ lib, buildPythonPackage, fetchPypi
2, pytest, pytestcov, pytest_xdist
3, six, numpy, scipy, pyyaml, h5py
4, keras-applications, keras-preprocessing
5}:
6
7buildPythonPackage rec {
8 pname = "Keras";
9 version = "2.4.3";
10
11 src = fetchPypi {
12 inherit pname version;
13 sha256 = "fedd729b52572fb108a98e3d97e1bac10a81d3917d2103cc20ab2a5f03beb973";
14 };
15
16 checkInputs = [
17 pytest
18 pytestcov
19 pytest_xdist
20 ];
21
22 propagatedBuildInputs = [
23 six pyyaml numpy scipy h5py
24 keras-applications keras-preprocessing
25 ];
26
27 # Couldn't get tests working
28 doCheck = false;
29
30 meta = with lib; {
31 description = "Deep Learning library for Theano and TensorFlow";
32 homepage = "https://keras.io";
33 license = licenses.mit;
34 maintainers = with maintainers; [ NikolaMandic ];
35 };
36}