pkgs/development/python-modules/dask/default.nix

   1 { lib
   2 , bokeh
   3 , buildPythonPackage
   4 , fetchFromGitHub
   5 , fsspec
   6 , pytestCheckHook
   7 , pytest-rerunfailures
   8 , pythonOlder
   9 , cloudpickle
  10 , numpy
  11 , toolz
  12 , dill
  13 , pandas
  14 , partd
  15 , pytest-xdist
  16 , withExtraComplete ? false
  17 , distributed
  18 }:
  19
  20 buildPythonPackage rec {
  21   pname = "dask";
  22   version = "2021.03.0";
  23   disabled = pythonOlder "3.5";
  24
  25   src = fetchFromGitHub {
  26     owner = "dask";
  27     repo = pname;
  28     rev = version;
  29     sha256 = "LACv7lWpQULQknNGX/9vH9ckLsypbqKDGnsNBgKT1eI=";
  30   };
  31
  32   propagatedBuildInputs = [
  33     bokeh
  34     cloudpickle
  35     dill
  36     fsspec
  37     numpy
  38     pandas
  39     partd
  40     toolz
  41   ] ++ lib.optionals withExtraComplete [
  42     distributed
  43   ];
  44
  45   doCheck = false;
  46
  47   checkInputs = [
  48     pytestCheckHook
  49     pytest-rerunfailures
  50     pytest-xdist
  51   ];
  52
  53   dontUseSetuptoolsCheck = true;
  54
  55   postPatch = ''
  56     # versioneer hack to set version of github package
  57     echo "def get_versions(): return {'dirty': False, 'error': None, 'full-revisionid': None, 'version': '${version}'}" > dask/_version.py
  58
  59     substituteInPlace setup.py \
  60       --replace "version=versioneer.get_version()," "version='${version}'," \
  61       --replace "cmdclass=versioneer.get_cmdclass()," ""
  62   '';
  63
  64   pytestFlagsArray = [ "-n $NIX_BUILD_CORES" ];
  65
  66   disabledTests = [
  67     "test_annotation_pack_unpack"
  68     "test_annotations_blockwise_unpack"
  69   ];
  70
  71   meta = with lib; {
  72     description = "Minimal task scheduling abstraction";
  73     homepage = "https://dask.org/";
  74     changelog = "https://docs.dask.org/en/latest/changelog.html";
  75     license = licenses.bsd3;
  76     maintainers = with maintainers; [ fridh ];
  77   };
  78 }