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