pkgs/development/python-modules/keras-preprocessing/default.nix

   1 { lib, buildPythonPackage, fetchPypi, numpy, six, scipy, pillow, pytest, keras }:
   2
   3 buildPythonPackage rec {
   4   pname = "Keras_Preprocessing";
   5   version = "1.1.2";
   6
   7   src = fetchPypi {
   8     inherit pname version;
   9     sha256 = "add82567c50c8bc648c14195bf544a5ce7c1f76761536956c3d2978970179ef3";
  10   };
  11
  12   propagatedBuildInputs = [
  13     # required
  14     numpy six
  15     # optional
  16     scipy pillow
  17   ];
  18
  19   nativeCheckInputs = [
  20     pytest keras
  21   ];
  22
  23   checkPhase = ''
  24     py.test tests/
  25   '';
  26
  27   # Cyclic dependency: keras-preprocessing's tests require Keras, which requires keras-preprocessing
  28   doCheck = false;
  29
  30   meta = with lib; {
  31     description = "Easy data preprocessing and data augmentation for deep learning models";
  32     homepage = "https://github.com/keras-team/keras-preprocessing";
  33     license = licenses.mit;
  34   };
  35 }