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