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.0.0";
13 pname = "dask-mpi";
14
15 src = fetchPypi {
16 inherit pname version;
17 sha256 = "774cd2d69e5f7154e1fa133c22498062edd31507ffa2ea19f4ab4d8975c27bc3";
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}