diff --git a/pkgs/development/python-modules/dask-ml/default.nix b/pkgs/development/python-modules/dask-ml/default.nix index 6b96c3d2199a..427f1b446c21 100644 --- a/pkgs/development/python-modules/dask-ml/default.nix +++ b/pkgs/development/python-modules/dask-ml/default.nix @@ -1,12 +1,15 @@ { lib, + stdenv, buildPythonPackage, - pythonOlder, fetchFromGitHub, + + # build-system hatch-vcs, hatchling, setuptools-scm, - dask, + + # dependencies dask-expr, dask-glm, distributed, @@ -17,6 +20,9 @@ pandas, scikit-learn, scipy, + dask, + + # tests pytest-mock, pytestCheckHook, }: @@ -26,8 +32,6 @@ buildPythonPackage rec { version = "2024.4.4"; pyproject = true; - disabled = pythonOlder "3.6"; - src = fetchFromGitHub { owner = "dask"; repo = "dask-ml"; @@ -66,11 +70,19 @@ buildPythonPackage rec { pytestCheckHook ]; - disabledTestPaths = [ - # AttributeError: 'csr_matrix' object has no attribute 'A' - # Fixed in https://github.com/dask/dask-ml/pull/996 - "tests/test_svd.py" - ]; + disabledTestPaths = + [ + # AttributeError: 'csr_matrix' object has no attribute 'A' + # Fixed in https://github.com/dask/dask-ml/pull/996 + "tests/test_svd.py" + ] + ++ lib.optionals stdenv.isDarwin [ + # RuntimeError: Not enough arguments provided: missing keys + "tests/model_selection/test_hyperband.py" + "tests/model_selection/test_incremental.py" + "tests/model_selection/test_incremental_warns.py" + "tests/model_selection/test_successive_halving.py" + ]; disabledTests = [ # Flaky: `Arrays are not almost equal to 3 decimals` (although values do actually match)