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; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 5ef8626d72ed..9b98a1af7891 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -16809,22 +16809,7 @@ EOF cudaSupport = true; }; - tflearn = buildPythonPackage rec { - name = "tflearn-0.2.1"; - - meta = { - description = "Deep learning library featuring a higher-level API for TensorFlow"; - homepage = "https://github.com/tflearn/tflearn"; - license = licenses.mit; - }; - - propagatedBuildInputs = with self; [ scipy h5py pillow tensorflow ]; - - src = pkgs.fetchurl { - url = "mirror://pypi/t/tflearn/${name}.tar.gz"; - sha256 = "1n884c4j35409id2bncyj5fvmmfpdqj3pk6wrv0s1znnvs0lkii0"; - }; - }; + tflearn = callPackage ../development/python-modules/tflearn { }; simpleai = buildPythonPackage rec { version = "0.7.11";