diff options
author | Chris Ostrouchov <chris.ostrouchov@gmail.com> | 2018-09-13 15:12:21 -0400 |
---|---|---|
committer | Chris Ostrouchov <chris.ostrouchov@gmail.com> | 2018-09-13 16:40:58 -0400 |
commit | 93cdaaa3d4f198420624495d64ba5b6af01c9e07 (patch) | |
tree | 58c9ae49bb4a5c20263d0f82b563140063268164 /pkgs/development | |
parent | e5cd420e7384fa936e9786a02201ab400b4dcca3 (diff) | |
download | nixlib-93cdaaa3d4f198420624495d64ba5b6af01c9e07.tar nixlib-93cdaaa3d4f198420624495d64ba5b6af01c9e07.tar.gz nixlib-93cdaaa3d4f198420624495d64ba5b6af01c9e07.tar.bz2 nixlib-93cdaaa3d4f198420624495d64ba5b6af01c9e07.tar.lz nixlib-93cdaaa3d4f198420624495d64ba5b6af01c9e07.tar.xz nixlib-93cdaaa3d4f198420624495d64ba5b6af01c9e07.tar.zst nixlib-93cdaaa3d4f198420624495d64ba5b6af01c9e07.zip |
pythonPackages.dask-ml: init at 0.10.0
Diffstat (limited to 'pkgs/development')
-rw-r--r-- | pkgs/development/python-modules/dask-ml/default.nix | 43 |
1 files changed, 43 insertions, 0 deletions
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 000000000000..0f53bcc225df --- /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 ]; + }; +} |