1{ lib
2, buildPythonPackage
3, fetchPypi
4, dask
5, distributed
6, mpi4py
7}:
8
9buildPythonPackage 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}