diff options
Diffstat (limited to 'pkgs/development/python-modules/dask-ml/default.nix')
| -rw-r--r-- | pkgs/development/python-modules/dask-ml/default.nix | 33 |
1 files changed, 27 insertions, 6 deletions
diff --git a/pkgs/development/python-modules/dask-ml/default.nix b/pkgs/development/python-modules/dask-ml/default.nix index e7e40f7777b..06c5f1753d4 100644 --- a/pkgs/development/python-modules/dask-ml/default.nix +++ b/pkgs/development/python-modules/dask-ml/default.nix @@ -1,6 +1,7 @@ { stdenv , buildPythonPackage , fetchPypi +, pythonOlder , dask , numpy, toolz # dask[array] , numba @@ -19,21 +20,41 @@ }: buildPythonPackage rec { - version = "1.0.0"; + version = "1.1.1"; pname = "dask-ml"; + disabled = pythonOlder "3.6"; # >= 3.6 src = fetchPypi { inherit pname version; - sha256 = "dde926478653bd03a3fbc501d3873a1534836608217b94d04320d1e1c07e59dc"; + sha256 = "1yad9b9hd02rbdf6m1gfj74cjgcbyp7lxdj22d5h5rhv7xa3127x"; }; - checkInputs = [ pytest xgboost tensorflow joblib distributed ]; - propagatedBuildInputs = [ dask numpy toolz numba pandas scikitlearn scipy dask-glm six multipledispatch packaging ]; + propagatedBuildInputs = [ + dask + dask-glm + distributed + multipledispatch + numba + numpy + packaging + pandas + scikitlearn + scipy + six + toolz + ]; - # dask-ml has some heavy test requirements - # and requires some very new packages + # 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"; |