pyrox.dev
1{ lib
2, buildPythonPackage
3, fetchPypi
4, tensorflow
5, absl-py
6, dm-sonnet
7, networkx
8, numpy
9, setuptools
10, six
11, future
12}:
13
14buildPythonPackage rec {
15 pname = "graph_nets";
16 version = "1.1.0";
17
18 src = fetchPypi {
19 inherit pname version;
20 sha256 = "278a040674bef295aaf8bb5b0d1b3f207144dc68f0bcfe3f14912b9b85eb0927";
21 };
22
23 postPatch = ''
24 # https://github.com/deepmind/graph_nets/issues/63
25 sed -i 's/dm-sonnet==1.23/dm-sonnet/' setup.py
26 '';
27
28 propagatedBuildInputs = [
29 tensorflow
30 absl-py
31 dm-sonnet
32 networkx
33 numpy
34 setuptools
35 six
36 future
37 ];
38
39 meta = with lib; {
40 description = "Build Graph Nets in Tensorflow";
41 homepage = "https://github.com/deepmind/graph_nets";
42 license = licenses.asl20;
43 maintainers = with maintainers; [ timokau ];
44 };
45}