tjh.dev
1{ stdenv
2, buildPythonPackage
3, fetchPypi
4, dask
5, distributed
6, mpi4py
7, pytest
8, requests
9}:
10
11buildPythonPackage rec {
12 version = "1.0.2";
13 pname = "dask-mpi";
14
15 src = fetchPypi {
16 inherit pname version;
17 sha256 = "1apzzh90gl9jx43z0gjmgpniplhvqziafi2l8688a0g01vw7ibjv";
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}