61 lines
1.3 KiB
Nix
61 lines
1.3 KiB
Nix
{ buildPythonPackage
|
|
, pytestCheckHook
|
|
, cmake
|
|
, scipy
|
|
, scikit-learn
|
|
, stdenv
|
|
, xgboost
|
|
, substituteAll
|
|
, pandas
|
|
, matplotlib
|
|
, graphviz
|
|
, datatable
|
|
, hypothesis
|
|
}:
|
|
|
|
buildPythonPackage {
|
|
pname = "xgboost";
|
|
inherit (xgboost) version src meta;
|
|
|
|
nativeBuildInputs = [ cmake ];
|
|
buildInputs = [ xgboost ];
|
|
propagatedBuildInputs = [ scipy ];
|
|
checkInputs = [
|
|
pytestCheckHook
|
|
scikit-learn
|
|
pandas
|
|
matplotlib
|
|
graphviz
|
|
datatable
|
|
hypothesis
|
|
];
|
|
|
|
# Override existing logic for locating libxgboost.so which is not appropriate for Nix
|
|
prePatch = let
|
|
libPath = "${xgboost}/lib/libxgboost${stdenv.hostPlatform.extensions.sharedLibrary}";
|
|
in ''
|
|
echo 'find_lib_path = lambda: ["${libPath}"]' > python-package/xgboost/libpath.py
|
|
'';
|
|
|
|
dontUseCmakeConfigure = true;
|
|
|
|
postPatch = ''
|
|
cd python-package
|
|
'';
|
|
|
|
preCheck = ''
|
|
ln -sf ../demo .
|
|
ln -s ${xgboost}/bin/xgboost ../xgboost
|
|
'';
|
|
|
|
pytestFlagsArray = ["../tests/python"];
|
|
disabledTestPaths = [
|
|
# Requires internet access: https://github.com/dmlc/xgboost/blob/03cd087da180b7dff21bd8ef34997bf747016025/tests/python/test_ranking.py#L81
|
|
"../tests/python/test_ranking.py"
|
|
];
|
|
disabledTests = [
|
|
"test_cli_binary_classification"
|
|
"test_model_compatibility"
|
|
];
|
|
|
|
}
|