1{ lib
2, buildPythonPackage
3, pkgs
4, requests
5, numpy
6, graphviz
7, python
8, isPy3k
9}:
10
11buildPythonPackage {
12 inherit (pkgs.mxnet) pname version src meta;
13
14 buildInputs = [ pkgs.mxnet ];
15 propagatedBuildInputs = [ requests numpy graphviz ];
16
17 LD_LIBRARY_PATH = lib.makeLibraryPath [ pkgs.mxnet ];
18
19 doCheck = !isPy3k;
20
21 postPatch = ''
22 substituteInPlace python/setup.py \
23 --replace "graphviz<0.9.0," "graphviz"
24 '';
25
26 preConfigure = ''
27 cd python
28 '';
29
30 postInstall = ''
31 rm -rf $out/mxnet
32 ln -s ${pkgs.mxnet}/lib/libmxnet.so $out/${python.sitePackages}/mxnet
33 '';
34
35}