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

   1 { lib
   2 , buildPythonPackage
   3 , fetchPypi
   4 , dask
   5 , distributed
   6 , mpi4py
   7 }:
   8
   9 buildPythonPackage rec {
  10   version = "2.21.0";
  11   pname = "dask-mpi";
  12
  13   src = fetchPypi {
  14     inherit pname version;
  15     sha256 = "76e153fc8c58047d898970b33ede0ab1990bd4e69cc130c6627a96f11b12a1a7";
  16   };
  17
  18   propagatedBuildInputs = [ dask distributed mpi4py ];
  19
  20   # hardcoded mpirun path in tests
  21   doCheck = false;
  22   pythonImportsCheck = [ "dask_mpi" ];
  23
  24   meta = with lib; {
  25     homepage = "https://github.com/dask/dask-mpi";
  26     description = "Deploy Dask using mpi4py";
  27     license = licenses.bsd3;
  28     maintainers = [ maintainers.costrouc ];
  29   };
  30 }