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