diff --git a/pkgs/development/python-modules/dm-sonnet/default.nix b/pkgs/development/python-modules/dm-sonnet/default.nix new file mode 100644 index 000000000000..f857eac8503b --- /dev/null +++ b/pkgs/development/python-modules/dm-sonnet/default.nix @@ -0,0 +1,54 @@ +{ lib +, fetchFromGitHub +, buildPythonPackage +, numpy +, tabulate +, six +, dm-tree +, absl-py +, wrapt +, docutils +, tensorflow +, tensorflow-datasets }: + +buildPythonPackage rec { + pname = "dm-sonnet"; + version = "2.0.0"; + + src = fetchFromGitHub { + owner = "deepmind"; + repo = "sonnet"; + rev = "v${version}"; + sha256 = "sha256-YSMeH5ZTfP1OdLBepsxXAVczBG/ghSjCWjoz/I+TFl8="; + }; + + buildInputs = [ + absl-py + dm-tree + numpy + six + tabulate + wrapt + ]; + + propagatedBuildInputs = [ + tabulate + tensorflow + ]; + + checkInputs = [ + docutils + tensorflow-datasets + ]; + + pythonImportsCheck = [ + "sonnet" + ]; + + meta = with lib; { + description = "Library for building neural networks in TensorFlow"; + homepage = "https://github.com/deepmind/sonnet"; + license = licenses.asl20; + maintainers = with maintainers; [ onny ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 37a003638e2e..2231b5cfa889 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -2527,6 +2527,8 @@ in { dm-haiku = callPackage ../development/python-modules/dm-haiku { }; + dm-sonnet = callPackage ../development/python-modules/dm-sonnet { }; + dm-tree = callPackage ../development/python-modules/dm-tree { abseil-cpp = pkgs.abseil-cpp.override { cxxStandard = "14";