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

   1 { lib, buildPythonPackage, fetchPypi
   2 , pytest, pytest-cov, pytest-xdist
   3 , six, numpy, scipy, pyyaml, h5py
   4 , keras-applications, keras-preprocessing
   5 }:
   6
   7 buildPythonPackage rec {
   8   pname = "keras";
   9   version = "2.10.0";
  10   format = "wheel";
  11
  12   src = fetchPypi {
  13     inherit format pname version;
  14     sha256 = "sha256-JqbiwlIudGjd6iJxCpmzKQSTdo/AijnnXRFzoONFL98=";
  15   };
  16
  17   checkInputs = [
  18     pytest
  19     pytest-cov
  20     pytest-xdist
  21   ];
  22
  23   propagatedBuildInputs = [
  24     six pyyaml numpy scipy h5py
  25     keras-applications keras-preprocessing
  26   ];
  27
  28   # Couldn't get tests working
  29   doCheck = false;
  30
  31   meta = with lib; {
  32     description = "Deep Learning library for Theano and TensorFlow";
  33     homepage = "https://keras.io";
  34     license = licenses.mit;
  35     maintainers = with maintainers; [ NikolaMandic ];
  36   };
  37 }