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