tjh.dev
1{
2 lib,
3 buildPythonPackage,
4 pkgs,
5 setuptools,
6 distutils,
7 requests,
8 numpy,
9 graphviz,
10 python,
11 isPy3k,
12 isPy310,
13}:
14
15buildPythonPackage {
16 inherit (pkgs.mxnet) pname version src;
17 pyproject = true;
18
19 build-system = [ setuptools ];
20
21 buildInputs = [ pkgs.mxnet ];
22
23 dependencies = [
24 distutils
25 requests
26 numpy
27 graphviz
28 ];
29
30 pythonRelaxDeps = [
31 "graphviz"
32 ];
33
34 LD_LIBRARY_PATH = lib.makeLibraryPath [ pkgs.mxnet ];
35
36 doCheck = !isPy3k;
37
38 postPatch = ''
39 # Required to support numpy >=1.24 where np.bool is removed in favor of just bool
40 substituteInPlace python/mxnet/numpy/utils.py \
41 --replace-fail "bool = onp.bool" "bool = bool"
42 '';
43
44 preConfigure = ''
45 cd python
46 '';
47
48 postInstall = ''
49 rm -rf $out/mxnet
50 ln -s ${pkgs.mxnet}/lib/libmxnet.so $out/${python.sitePackages}/mxnet
51 '';
52
53 meta = pkgs.mxnet.meta // {
54 broken = (pkgs.mxnet.broken or false) || (isPy310 && pkgs.mxnet.cudaSupport);
55 };
56}