diff options
author | Jonas Heinrich <onny@project-insanity.org> | 2022-07-14 17:55:04 +0200 |
---|---|---|
committer | Yt <happysalada@proton.me> | 2022-07-16 15:02:05 -0400 |
commit | e4983ba583b370ce944904efe5183b451d4092bc (patch) | |
tree | c7f4ab66d2d4d55fc9c70fb42a3125767b58f88c /pkgs/development | |
parent | 6d2e664e8fbd1783c2fedaa2c717c91fa6dd5333 (diff) |
python310Packages.dm-sonnet: init at 2.0.0
Diffstat (limited to 'pkgs/development')
-rw-r--r-- | pkgs/development/python-modules/dm-sonnet/default.nix | 54 |
1 files changed, 54 insertions, 0 deletions
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 0000000000000..f857eac8503b7 --- /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 ]; + }; +} |