56 lines
979 B
Nix
56 lines
979 B
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
pkgs,
|
|
setuptools,
|
|
distutils,
|
|
requests,
|
|
numpy,
|
|
graphviz,
|
|
python,
|
|
isPy3k,
|
|
isPy310,
|
|
}:
|
|
|
|
buildPythonPackage {
|
|
inherit (pkgs.mxnet) pname version src;
|
|
pyproject = true;
|
|
|
|
build-system = [ setuptools ];
|
|
|
|
buildInputs = [ pkgs.mxnet ];
|
|
|
|
dependencies = [
|
|
distutils
|
|
requests
|
|
numpy
|
|
graphviz
|
|
];
|
|
|
|
pythonRelaxDeps = [
|
|
"graphviz"
|
|
];
|
|
|
|
LD_LIBRARY_PATH = lib.makeLibraryPath [ pkgs.mxnet ];
|
|
|
|
doCheck = !isPy3k;
|
|
|
|
postPatch = ''
|
|
# Required to support numpy >=1.24 where np.bool is removed in favor of just bool
|
|
substituteInPlace python/mxnet/numpy/utils.py \
|
|
--replace-fail "bool = onp.bool" "bool = bool"
|
|
'';
|
|
|
|
preConfigure = ''
|
|
cd python
|
|
'';
|
|
|
|
postInstall = ''
|
|
rm -rf $out/mxnet
|
|
ln -s ${pkgs.mxnet}/lib/libmxnet.so $out/${python.sitePackages}/mxnet
|
|
'';
|
|
|
|
meta = pkgs.mxnet.meta // {
|
|
broken = (pkgs.mxnet.broken or false) || (isPy310 && pkgs.mxnet.cudaSupport);
|
|
};
|
|
}
|