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

   1 { lib
   2 , buildPythonPackage
   3 , fetchPypi
   4 , fetchpatch
   5 , dask
   6 , distributed
   7 , mpi4py
   8 , pythonOlder
   9 }:
  10
  11 buildPythonPackage rec {
  12   pname = "dask-mpi";
  13   version = "2022.4.0";
  14   format = "setuptools";
  15
  16   disabled = pythonOlder "3.6";
  17
  18   src = fetchPypi {
  19     inherit pname version;
  20     hash = "sha256-CgTx19NaBs3/UGWTMw1EFOokLJFySYzhkfV0LqxJnhc=";
  21   };
  22
  23   patches = [
  24     # https://github.com/dask/dask-mpi/pull/123
  25     (fetchpatch {
  26       name = "fix-versioneer-on-python312.patch";
  27       url = "https://github.com/dask/dask-mpi/pull/123/commits/0f3b0286b7e29b5d5475561a148dc398108fc259.patch";
  28       hash = "sha256-xXADCSIhq1ARny2twzrhR1J8LkMFWFl6tmGxrM8RvkU=";
  29     })
  30   ];
  31
  32   propagatedBuildInputs = [
  33     dask
  34     distributed
  35     mpi4py
  36   ];
  37
  38   # Hardcoded mpirun path in tests
  39   doCheck = false;
  40
  41   pythonImportsCheck = [
  42     "dask_mpi"
  43   ];
  44
  45   meta = with lib; {
  46     description = "Deploy Dask using mpi4py";
  47     mainProgram = "dask-mpi";
  48     homepage = "https://github.com/dask/dask-mpi";
  49     license = licenses.bsd3;
  50     maintainers = with maintainers; [ ];
  51   };
  52 }