nixpkgs/pkgs/development/python-modules/tflearn/default.nix
2020-12-01 14:44:21 +01:00

25 lines
598 B
Nix

{ lib, fetchPypi, buildPythonPackage, pytest, scipy, h5py
, pillow, tensorflow }:
buildPythonPackage rec {
pname = "tflearn";
version = "0.5.0";
src = fetchPypi {
inherit pname version;
sha256 = "818aa57667693810415dc203ba3f75f1541e931a8dc30b6e8b21563541a70388";
};
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;
};
}