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

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