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

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchPypi,
   5   numpy,
   6   h5py,
   7 }:
   8
   9 buildPythonPackage rec {
  10   pname = "keras-applications";
  11   version = "1.0.8";
  12
  13   src = fetchPypi {
  14     pname = "Keras_Applications";
  15     inherit version;
  16     sha256 = "5579f9a12bcde9748f4a12233925a59b93b73ae6947409ff34aa2ba258189fe5";
  17   };
  18
  19   # Cyclic dependency: keras-applications requires keras, which requires keras-applications
  20   postPatch = ''
  21     sed -i "s/keras>=[^']*//" setup.py
  22   '';
  23
  24   # No tests in PyPI tarball
  25   doCheck = false;
  26
  27   propagatedBuildInputs = [
  28     numpy
  29     h5py
  30   ];
  31
  32   meta = with lib; {
  33     description = "Reference implementations of popular deep learning models";
  34     homepage = "https://github.com/keras-team/keras-applications";
  35     license = licenses.mit;
  36   };
  37 }