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

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