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