tjh.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.0.4";
17
18 src = fetchPypi {
19 inherit pname version;
20 sha256 = "6f27f0358a477d4ec3ab300bf09944b4dfdd5c10b0a5f7ac06bcb32d1327ebe1";
21 };
22
23 buildInputs = [];
24
25 postPatch = ''
26 # https://github.com/deepmind/graph_nets/issues/63
27 sed -i 's/dm-sonnet==1.23/dm-sonnet/' setup.py
28 '';
29
30 propagatedBuildInputs = [
31 tensorflow
32 absl-py
33 dm-sonnet
34 networkx
35 numpy
36 setuptools
37 six
38 future
39 ];
40
41 meta = with lib; {
42 description = "Build Graph Nets in Tensorflow";
43 homepage = https://github.com/deepmind/graph_nets;
44 license = licenses.asl20;
45 maintainers = with maintainers; [ timokau ];
46 };
47}