59 lines
1.1 KiB
Nix
59 lines
1.1 KiB
Nix
|
{
|
||
|
|
buildPythonPackage,
|
||
|
|
dlib,
|
||
|
|
pytestCheckHook,
|
||
|
|
more-itertools,
|
||
|
|
}:
|
||
|
|
|
||
|
|
buildPythonPackage {
|
||
|
|
inherit (dlib)
|
||
|
|
stdenv
|
||
|
|
pname
|
||
|
|
version
|
||
|
|
src
|
||
|
|
nativeBuildInputs
|
||
|
|
buildInputs
|
||
|
|
cmakeFlags
|
||
|
|
passthru
|
||
|
|
meta
|
||
|
|
;
|
||
|
|
|
||
|
|
format = "setuptools";
|
||
|
|
|
||
|
|
patches = [ ./build-cores.patch ];
|
||
|
|
|
||
|
|
nativeCheckInputs = [
|
||
|
|
pytestCheckHook
|
||
|
|
more-itertools
|
||
|
|
];
|
||
|
|
|
||
|
|
postPatch = ''
|
||
|
|
substituteInPlace setup.py \
|
||
|
|
--replace "more-itertools<6.0.0" "more-itertools" \
|
||
|
|
--replace "pytest==3.8" "pytest"
|
||
|
|
'';
|
||
|
|
|
||
|
|
# Pass CMake flags through to the build script
|
||
|
|
preConfigure = ''
|
||
|
|
for flag in $cmakeFlags; do
|
||
|
|
if [[ "$flag" == -D* ]]; then
|
||
|
|
setupPyBuildFlags+=" --set ''${flag#-D}"
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
'';
|
||
|
|
|
||
|
|
dontUseCmakeConfigure = true;
|
||
|
|
|
||
|
|
doCheck =
|
||
|
|
!(
|
||
|
|
# The tests attempt to use CUDA on the build platform.
|
||
|
|
# https://github.com/NixOS/nixpkgs/issues/225912
|
||
|
|
dlib.cudaSupport
|
||
|
|
|
||
|
|
# although AVX can be enabled, we never test with it. Some Hydra machines
|
||
|
|
# fail because of this, however their build results are probably used on hardware
|
||
|
|
# with AVX support.
|
||
|
|
|| dlib.avxSupport
|
||
|
|
);
|
||
|
|
}
|