diff --git a/pkgs/development/python-modules/dask/default.nix b/pkgs/development/python-modules/dask/default.nix index 59869efbffd8..d0662536538f 100644 --- a/pkgs/development/python-modules/dask/default.nix +++ b/pkgs/development/python-modules/dask/default.nix @@ -4,6 +4,7 @@ , buildPythonPackage , cloudpickle , distributed +, fastparquet , fetchFromGitHub , fetchpatch , fsspec @@ -12,17 +13,19 @@ , packaging , pandas , partd +, pyarrow , pytest-rerunfailures , pytest-xdist , pytestCheckHook , pythonOlder , pyyaml +, scipy , toolz }: buildPythonPackage rec { pname = "dask"; - version = "2022.02.1"; + version = "2022.05.2"; format = "setuptools"; disabled = pythonOlder "3.7"; @@ -31,7 +34,7 @@ buildPythonPackage rec { owner = "dask"; repo = pname; rev = version; - hash = "sha256-A8ktvfpow/QKAEEt9SUnkTqYFJCrV1mgnuDIP3gdyrE="; + hash = "sha256-8M70Pf61PhYnBPRhSG55eWg6gK0lxsIFKF+cRCsf0/U="; }; propagatedBuildInputs = [ @@ -41,48 +44,60 @@ buildPythonPackage rec { partd pyyaml toolz - pandas - jinja2 - bokeh - numpy ]; - doCheck = true; + passthru.optional-dependencies = { + array = [ + numpy + ]; + dataframe = [ + numpy + pandas + ]; + distributed = [ + distributed + ]; + diagnostics = [ + bokeh + jinja2 + ]; + }; checkInputs = [ + fastparquet + pyarrow pytestCheckHook pytest-rerunfailures pytest-xdist + scipy ]; dontUseSetuptoolsCheck = true; postPatch = '' - # versioneer hack to set version of github package + # versioneer hack to set version of GitHub package echo "def get_versions(): return {'dirty': False, 'error': None, 'full-revisionid': None, 'version': '${version}'}" > dask/_version.py substituteInPlace setup.py \ --replace "version=versioneer.get_version()," "version='${version}'," \ --replace "cmdclass=versioneer.get_cmdclass()," "" + + substituteInPlace setup.cfg \ + --replace " --durations=10" "" \ + --replace " -v" "" ''; pytestFlagsArray = [ - # rerun failed tests up to three times + # Rerun failed tests up to three times "--reruns 3" - # don't run tests that require network access + # Don't run tests that require network access "-m 'not network'" ]; disabledTests = lib.optionals stdenv.isDarwin [ - # this test requires features of python3Packages.psutil that are + # Test requires features of python3Packages.psutil that are # blocked in sandboxed-builds "test_auto_blocksize_csv" - ] ++ [ - # A deprecation warning from newer sqlalchemy versions makes these tests - # to fail https://github.com/dask/dask/issues/7406 - "test_sql" - # Test interrupt fails intermittently https://github.com/dask/dask/issues/2192 - "test_interrupt" ]; __darwinAllowLocalNetworking = true;