pkgs/development/python-modules/tflearn/default.nix

   1 { lib, fetchPypi, buildPythonPackage, pytest, scipy, h5py
   2 , pillow, tensorflow }:
   3
   4 buildPythonPackage rec {
   5   pname = "tflearn";
   6   version = "0.5.0";
   7
   8   src = fetchPypi {
   9     inherit pname version;
  10     sha256 = "818aa57667693810415dc203ba3f75f1541e931a8dc30b6e8b21563541a70388";
  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 }