blob: 5dfcd2f415b9df21f9e9b1b40565d247bbdb97b2 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
{ stdenv
, buildPythonPackage
, fetchPypi
, xgboost
, dask
, distributed
, pytest
, scikitlearn
}:
buildPythonPackage rec {
version = "0.1.5";
pname = "dask-xgboost";
src = fetchPypi {
inherit pname version;
sha256 = "1860d06965fe68def1c83b9195130a92050fd4bc28bf2be689898a3a74ee1316";
};
checkInputs = [ pytest scikitlearn ];
propagatedBuildInputs = [ xgboost dask distributed ];
checkPhase = ''
py.test dask_xgboost/tests/test_core.py
'';
doCheck = false;
meta = with stdenv.lib; {
homepage = https://github.com/dask/dask-xgboost;
description = "Interactions between Dask and XGBoost";
license = licenses.bsd3;
maintainers = [ maintainers.costrouc ];
};
}
|