diff options
Diffstat (limited to 'nixpkgs/pkgs/development/python-modules/dask-ml')
-rw-r--r-- | nixpkgs/pkgs/development/python-modules/dask-ml/default.nix | 64 |
1 files changed, 64 insertions, 0 deletions
diff --git a/nixpkgs/pkgs/development/python-modules/dask-ml/default.nix b/nixpkgs/pkgs/development/python-modules/dask-ml/default.nix new file mode 100644 index 000000000000..c5fafa495aa6 --- /dev/null +++ b/nixpkgs/pkgs/development/python-modules/dask-ml/default.nix @@ -0,0 +1,64 @@ +{ stdenv +, buildPythonPackage +, fetchPypi +, pythonOlder +, dask +, numpy, toolz # dask[array] +, numba +, pandas +, scikitlearn +, scipy +, dask-glm +, six +, multipledispatch +, packaging +, pytest +, xgboost +, tensorflow +, joblib +, distributed +}: + +buildPythonPackage rec { + version = "1.2.0"; + pname = "dask-ml"; + disabled = pythonOlder "3.6"; # >= 3.6 + + src = fetchPypi { + inherit pname version; + sha256 = "0ppg8licvkxz1af2q87cxms2p6ss2r5d4fdkbcivph56r0v0ci2k"; + }; + + propagatedBuildInputs = [ + dask + dask-glm + distributed + multipledispatch + numba + numpy + packaging + pandas + scikitlearn + scipy + six + toolz + ]; + + # has non-standard build from source, and pypi doesn't include tests + doCheck = false; + + # in lieu of proper tests + pythonImportsCheck = [ + "dask_ml" + "dask_ml.naive_bayes" + "dask_ml.wrappers" + "dask_ml.utils" + ]; + + meta = with stdenv.lib; { + homepage = https://github.com/dask/dask-ml; + description = "Scalable Machine Learn with Dask"; + license = licenses.bsd3; + maintainers = [ maintainers.costrouc ]; + }; +} |