pyrox.dev
1{
2 lib,
3 fetchPypi,
4 buildPythonPackage,
5 pytest,
6 scipy,
7 h5py,
8 pillow,
9 tensorflow,
10}:
11
12buildPythonPackage rec {
13 pname = "tflearn";
14 version = "0.5.0";
15 format = "setuptools";
16
17 src = fetchPypi {
18 inherit pname version;
19 sha256 = "818aa57667693810415dc203ba3f75f1541e931a8dc30b6e8b21563541a70388";
20 };
21
22 buildInputs = [ pytest ];
23
24 propagatedBuildInputs = [
25 scipy
26 h5py
27 pillow
28 tensorflow
29 ];
30
31 doCheck = false;
32
33 meta = with lib; {
34 description = "Deep learning library featuring a higher-level API for TensorFlow";
35 homepage = "https://github.com/tflearn/tflearn";
36 license = licenses.mit;
37 };
38}