diff --git a/pkgs/development/python-modules/graph_nets/default.nix b/pkgs/development/python-modules/graph_nets/default.nix new file mode 100644 index 000000000000..58aa6c84ade3 --- /dev/null +++ b/pkgs/development/python-modules/graph_nets/default.nix @@ -0,0 +1,47 @@ +{ lib +, buildPythonPackage +, fetchPypi +, tensorflow +, absl-py +, dm-sonnet +, networkx +, numpy +, setuptools +, six +, future +}: + +buildPythonPackage rec { + pname = "graph_nets"; + version = "1.0.3"; + + src = fetchPypi { + inherit pname version; + sha256 = "15cbs9smmgqz2n9mnlzdbqj3iv9iw179d2g0f9lnimdy7xl4jqdf"; + }; + + buildInputs = []; + + postPatch = '' + # https://github.com/deepmind/graph_nets/issues/63 + sed -i 's/dm-sonnet==1.23/dm-sonnet/' setup.py + ''; + + propagatedBuildInputs = [ + tensorflow + absl-py + dm-sonnet + networkx + numpy + setuptools + six + future + ]; + + meta = with lib; { + description = "Build Graph Nets in Tensorflow"; + homepage = https://github.com/deepmind/graph_nets; + license = licenses.asl20; + maintainers = with maintainers; [ timokau ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 37a3b376cf15..934aa1cb1c9e 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -4877,6 +4877,8 @@ in { graphite_beacon = callPackage ../development/python-modules/graphite_beacon { }; + graph_nets = callPackage ../development/python-modules/graph_nets { }; + influxgraph = callPackage ../development/python-modules/influxgraph { }; graphitepager = callPackage ../development/python-modules/graphitepager { };