xgboost: 0.72 -> 0.90

2019-06-03 18:18:03 +03:00
parent 1f65bb626d
commit e188f6765a
3 changed files with 9 additions and 24 deletions
--- a/pkgs/development/python-modules/xgboost/default.nix
+++ b/pkgs/development/python-modules/xgboost/default.nix
@@ -1,9 +1,15 @@
 { stdenv
 , buildPythonPackage
+, pytest
 , nose
 , scipy
+, scikitlearn
 , xgboost
 , substituteAll
+, pandas
+, matplotlib
+, graphviz
+, datatable
 }:

 buildPythonPackage rec {
@@ -21,7 +27,7 @@ buildPythonPackage rec {

  propagatedBuildInputs = [ scipy ];
  buildInputs = [ xgboost ];
-  checkInputs = [ nose ];
+  checkInputs = [ nose pytest scikitlearn pandas matplotlib graphviz datatable ];

  checkPhase = ''
    ln -sf ../demo .