tjh.dev
1{ stdenv
2, buildPythonPackage
3, fetchPypi
4, pythonOlder
5, dask
6, numpy, toolz # dask[array]
7, numba
8, pandas
9, scikitlearn
10, scipy
11, dask-glm
12, six
13, multipledispatch
14, packaging
15, pytest
16, xgboost
17, tensorflow
18, joblib
19, distributed
20}:
21
22buildPythonPackage rec {
23 version = "1.2.0";
24 pname = "dask-ml";
25 disabled = pythonOlder "3.6"; # >= 3.6
26
27 src = fetchPypi {
28 inherit pname version;
29 sha256 = "0ppg8licvkxz1af2q87cxms2p6ss2r5d4fdkbcivph56r0v0ci2k";
30 };
31
32 propagatedBuildInputs = [
33 dask
34 dask-glm
35 distributed
36 multipledispatch
37 numba
38 numpy
39 packaging
40 pandas
41 scikitlearn
42 scipy
43 six
44 toolz
45 ];
46
47 # has non-standard build from source, and pypi doesn't include tests
48 doCheck = false;
49
50 # in lieu of proper tests
51 pythonImportsCheck = [
52 "dask_ml"
53 "dask_ml.naive_bayes"
54 "dask_ml.wrappers"
55 "dask_ml.utils"
56 ];
57
58 meta = with stdenv.lib; {
59 homepage = https://github.com/dask/dask-ml;
60 description = "Scalable Machine Learn with Dask";
61 license = licenses.bsd3;
62 maintainers = [ maintainers.costrouc ];
63 };
64}