tjh.dev
1{ stdenv
2, buildPythonPackage
3, fetchPypi
4, dask
5, numpy, toolz # dask[array]
6, numba
7, pandas
8, scikitlearn
9, scipy
10, dask-glm
11, six
12, multipledispatch
13, packaging
14, pytest
15, xgboost
16, tensorflow
17, joblib
18, distributed
19}:
20
21buildPythonPackage rec {
22 version = "0.11.0";
23 pname = "dask-ml";
24
25 src = fetchPypi {
26 inherit pname version;
27 sha256 = "a9e8e69494560dc23534adb236e88b3b21dc30a156648453c9c6e4b27ff2df96";
28 };
29
30 checkInputs = [ pytest xgboost tensorflow joblib distributed ];
31 propagatedBuildInputs = [ dask numpy toolz numba pandas scikitlearn scipy dask-glm six multipledispatch packaging ];
32
33 # dask-ml has some heavy test requirements
34 # and requires some very new packages
35 doCheck = false;
36
37 meta = with stdenv.lib; {
38 homepage = https://github.com/dask/dask-ml;
39 description = "Scalable Machine Learn with Dask";
40 license = licenses.bsd3;
41 maintainers = [ maintainers.costrouc ];
42 };
43}