diff --git a/pkgs/development/python-modules/dask-ml/default.nix b/pkgs/development/python-modules/dask-ml/default.nix new file mode 100644 index 00000000000..0f53bcc225d --- /dev/null +++ b/pkgs/development/python-modules/dask-ml/default.nix @@ -0,0 +1,43 @@ +{ stdenv +, buildPythonPackage +, fetchPypi +, dask +, numpy, toolz # dask[array] +, numba +, pandas +, scikitlearn +, scipy +, dask-glm +, six +, multipledispatch +, packaging +, pytest +, xgboost +, tensorflow +, joblib +, distributed +}: + +buildPythonPackage rec { + version = "0.10.0"; + pname = "dask-ml"; + + src = fetchPypi { + inherit pname version; + sha256 = "4b6ca548c7282c1b6983e696e4bdfa0a2d7b51b168928b9322ea7a4b9a9f20f9"; + }; + + checkInputs = [ pytest xgboost tensorflow joblib distributed ]; + propagatedBuildInputs = [ dask numpy toolz numba pandas scikitlearn scipy dask-glm six multipledispatch packaging ]; + + # dask-ml has some heavy test requirements + # and requires some very new packages + doCheck = false; + + meta = with stdenv.lib; { + homepage = https://github.com/dask/dask-ml; + description = "Scalable Machine Learn with Dask"; + license = licenses.bsd3; + maintainers = [ maintainers.costrouc ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 4375ab550fd..bedfe6d4a78 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -1984,6 +1984,8 @@ in { dask = callPackage ../development/python-modules/dask { }; + dask-ml = callPackage ../development/python-modules/dask-ml { }; + datrie = callPackage ../development/python-modules/datrie { }; heapdict = callPackage ../development/python-modules/heapdict { };