diff options
Diffstat (limited to 'pkgs/development/python-modules/tflearn/default.nix')
-rw-r--r-- | pkgs/development/python-modules/tflearn/default.nix | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/tflearn/default.nix b/pkgs/development/python-modules/tflearn/default.nix new file mode 100644 index 000000000000..341c1da56801 --- /dev/null +++ b/pkgs/development/python-modules/tflearn/default.nix @@ -0,0 +1,24 @@ +{ lib, fetchPypi, buildPythonPackage, fetchurl, pytest, scipy, h5py +, pillow, tensorflow }: + +buildPythonPackage rec { + pname = "tflearn"; + version = "0.3.2"; + + src = fetchPypi { + inherit pname version; + sha256 = "034lvbslcivyj64r4w6xmr90ckmyxmrnkka9kal50x4175h02n1z"; + }; + + buildInputs = [ pytest ]; + + propagatedBuildInputs = [ scipy h5py pillow tensorflow ]; + + doCheck = false; + + meta = with lib; { + description = "Deep learning library featuring a higher-level API for TensorFlow"; + homepage = "https://github.com/tflearn/tflearn"; + license = licenses.mit; + }; +} |