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