36 lines
613 B
Nix
36 lines
613 B
Nix
{ stdenv
|
|
, buildPythonPackage
|
|
, pytest
|
|
, nose
|
|
, scipy
|
|
, scikitlearn
|
|
, xgboost
|
|
, substituteAll
|
|
, pandas
|
|
, matplotlib
|
|
, graphviz
|
|
, datatable
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "xgboost";
|
|
inherit (xgboost) version src meta;
|
|
|
|
patches = [
|
|
(substituteAll {
|
|
src = ./lib-path-for-python.patch;
|
|
libpath = "${xgboost}/lib";
|
|
})
|
|
];
|
|
|
|
postPatch = "cd python-package";
|
|
|
|
propagatedBuildInputs = [ scipy ];
|
|
buildInputs = [ xgboost ];
|
|
checkInputs = [ nose pytest scikitlearn pandas matplotlib graphviz datatable ];
|
|
|
|
checkPhase = ''
|
|
ln -sf ../demo .
|
|
nosetests ../tests/python
|
|
'';
|
|
}
|