diff options
Diffstat (limited to 'pkgs/development/python-modules/dask-ml/default.nix')
-rw-r--r-- | pkgs/development/python-modules/dask-ml/default.nix | 70 |
1 files changed, 42 insertions, 28 deletions
diff --git a/pkgs/development/python-modules/dask-ml/default.nix b/pkgs/development/python-modules/dask-ml/default.nix index 3cb7ed9618b15..2f6ffeff4ecd7 100644 --- a/pkgs/development/python-modules/dask-ml/default.nix +++ b/pkgs/development/python-modules/dask-ml/default.nix @@ -1,37 +1,48 @@ -{ lib -, buildPythonPackage -, dask -, dask-glm -, distributed -, fetchPypi -, multipledispatch -, numba -, numpy -, packaging -, pandas -, pythonOlder -, scikit-learn -, scipy -, setuptools-scm +{ + lib, + buildPythonPackage, + pythonOlder, + fetchFromGitHub, + hatch-vcs, + hatchling, + setuptools-scm, + dask, + dask-expr, + dask-glm, + distributed, + multipledispatch, + numba, + numpy, + packaging, + pandas, + scikit-learn, + scipy, + pytest-mock, + pytestCheckHook, }: buildPythonPackage rec { pname = "dask-ml"; - version = "2023.3.24"; - format = "setuptools"; + version = "2024.4.4"; + pyproject = true; disabled = pythonOlder "3.6"; - src = fetchPypi { - inherit pname version; - hash = "sha256-lsCQ220yg2U24/Ccpk3rWZ6GRYeqjj1NLGtK9YhzMwc="; + src = fetchFromGitHub { + owner = "dask"; + repo = "dask-ml"; + rev = "refs/tags/v${version}"; + hash = "sha256-ZiBpCk3b4Tk0Hwb4uapJLEx+Nb/qHFROCnkBTNGDzoU="; }; - nativeBuildInputs = [ + build-system = [ + hatch-vcs + hatchling setuptools-scm ]; - propagatedBuildInputs = [ + dependencies = [ + dask-expr dask-glm distributed multipledispatch @@ -41,11 +52,7 @@ buildPythonPackage rec { pandas scikit-learn scipy - ] ++ dask.optional-dependencies.array - ++ dask.optional-dependencies.dataframe; - - # has non-standard build from source, and pypi doesn't include tests - doCheck = false; + ] ++ dask.optional-dependencies.array ++ dask.optional-dependencies.dataframe; pythonImportsCheck = [ "dask_ml" @@ -54,10 +61,17 @@ buildPythonPackage rec { "dask_ml.utils" ]; + nativeCheckInputs = [ + pytest-mock + pytestCheckHook + ]; + + __darwinAllowLocalNetworking = true; + meta = with lib; { description = "Scalable Machine Learn with Dask"; homepage = "https://github.com/dask/dask-ml"; license = licenses.bsd3; - maintainers = with maintainers; [ ]; + maintainers = with maintainers; [ GaetanLepage ]; }; } |