pkgs/development/python-modules/dm-sonnet/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   click,
   5   dm-tree,
   6   docutils,
   7   etils,
   8   fetchFromGitHub,
   9   fetchpatch,
  10   numpy,
  11   pythonOlder,
  12   tabulate,
  13   tensorflow,
  14   tensorflow-datasets,
  15   wrapt,
  16 }:
  17
  18 buildPythonPackage rec {
  19   pname = "dm-sonnet";
  20   version = "2.0.2";
  21   format = "setuptools";
  22
  23   disabled = pythonOlder "3.6";
  24
  25   src = fetchFromGitHub {
  26     owner = "deepmind";
  27     repo = "sonnet";
  28     rev = "refs/tags/v${version}";
  29     hash = "sha256-WkloUbqSyPG3cbLG8ktsjdluACkCbUZ7t6rYWst8rs8=";
  30   };
  31
  32   patches = [
  33     (fetchpatch {
  34       name = "replace-np-bool-with-np-bool_.patch";
  35       url = "https://github.com/deepmind/sonnet/commit/df5d099d4557a9a81a0eb969e5a81ed917bcd612.patch";
  36       hash = "sha256-s7abl83osD4wa0ZhqgDyjqQ3gagwGYCdQifwFqhNp34=";
  37     })
  38   ];
  39
  40   propagatedBuildInputs = [
  41     dm-tree
  42     etils
  43     numpy
  44     tabulate
  45     wrapt
  46   ] ++ etils.optional-dependencies.epath;
  47
  48   optional-dependencies = {
  49     tensorflow = [ tensorflow ];
  50   };
  51
  52   nativeCheckInputs = [
  53     click
  54     docutils
  55     tensorflow
  56     tensorflow-datasets
  57   ];
  58
  59   pythonImportsCheck = [ "sonnet" ];
  60
  61   meta = with lib; {
  62     description = "Library for building neural networks in TensorFlow";
  63     homepage = "https://github.com/deepmind/sonnet";
  64     license = licenses.asl20;
  65     maintainers = with maintainers; [ onny ];
  66   };
  67 }