tjh.dev
1{ lib, fetchPypi, buildPythonPackage, pytest, scipy, h5py
2, pillow, tensorflow }:
3
4buildPythonPackage rec {
5 pname = "tflearn";
6 version = "0.3.2";
7
8 src = fetchPypi {
9 inherit pname version;
10 sha256 = "034lvbslcivyj64r4w6xmr90ckmyxmrnkka9kal50x4175h02n1z";
11 };
12
13 buildInputs = [ pytest ];
14
15 propagatedBuildInputs = [ scipy h5py pillow tensorflow ];
16
17 doCheck = false;
18
19 meta = with lib; {
20 description = "Deep learning library featuring a higher-level API for TensorFlow";
21 homepage = "https://github.com/tflearn/tflearn";
22 license = licenses.mit;
23 };
24}