tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
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.5";
17
18 src = fetchPypi {
19 inherit pname version;
20 sha256 = "73e98a5930c6c7b609257ada2cf4523db70810343e0593bf0fa710b77e991933";
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}