pkgs/development/python-modules/xgboost/default.nix

   1 { buildPythonPackage
   2 , pytest
   3 , nose
   4 , scipy
   5 , scikitlearn
   6 , stdenv
   7 , xgboost
   8 , substituteAll
   9 , pandas
  10 , matplotlib
  11 , graphviz
  12 , datatable
  13 }:
  14
  15 buildPythonPackage {
  16   pname = "xgboost";
  17   inherit (xgboost) version src meta;
  18
  19   patches = [
  20     (substituteAll {
  21       src = ./lib-path-for-python.patch;
  22       libpath = "${xgboost}/lib";
  23       extention = stdenv.hostPlatform.extensions.sharedLibrary;
  24     })
  25   ];
  26
  27   postPatch = "cd python-package";
  28
  29   propagatedBuildInputs = [ scipy ];
  30   buildInputs = [ xgboost ];
  31   checkInputs = [ nose pytest scikitlearn pandas matplotlib graphviz datatable ];
  32
  33   checkPhase = ''
  34     ln -sf ../demo .
  35     nosetests ../tests/python
  36   '';
  37 }