pkgs/development/python-modules/mpi4py/default.nix

   1 { lib, fetchPypi, fetchpatch, python, buildPythonPackage, mpi, openssh }:
   2
   3 buildPythonPackage rec {
   4   pname = "mpi4py";
   5   version = "3.0.3";
   6
   7   src = fetchPypi {
   8     inherit pname version;
   9     sha256 = "012d716c8b9ed1e513fcc4b18e5af16a8791f51e6d1716baccf988ad355c5a1f";
  10   };
  11
  12   patches = [ (fetchpatch {
  13     name = "disable-broken-test"; # upstream patch
  14     url = "https://github.com/mpi4py/mpi4py/commit/e13cc3ee59ec6ec2c6ee20e384e1e649d5027e8a.patch";
  15     sha256 = "0iwknrhxnfmsqjj8ahpn50c8pcdyv9p3wmcqi1jhr4i5y7lnmvvx";
  16   })];
  17
  18   passthru = {
  19     inherit mpi;
  20   };
  21
  22   postPatch = ''
  23     substituteInPlace test/test_spawn.py --replace \
  24                       "unittest.skipMPI('openmpi(<3.0.0)')" \
  25                       "unittest.skipMPI('openmpi')"
  26   '';
  27
  28   configurePhase = "";
  29
  30   installPhase = ''
  31     mkdir -p "$out/lib/${python.libPrefix}/site-packages"
  32     export PYTHONPATH="$out/lib/${python.libPrefix}/site-packages:$PYTHONPATH"
  33
  34     ${python}/bin/${python.executable} setup.py install \
  35       --install-lib=$out/lib/${python.libPrefix}/site-packages \
  36       --prefix="$out"
  37
  38     # --install-lib:
  39     # sometimes packages specify where files should be installed outside the usual
  40     # python lib prefix, we override that back so all infrastructure (setup hooks)
  41     # work as expected
  42
  43     # Needed to run the tests reliably. See:
  44     # https://bitbucket.org/mpi4py/mpi4py/issues/87/multiple-test-errors-with-openmpi-30
  45     export OMPI_MCA_rmaps_base_oversubscribe=yes
  46   '';
  47
  48   setupPyBuildFlags = ["--mpicc=${mpi}/bin/mpicc"];
  49
  50   nativeBuildInputs = [ mpi openssh ];
  51
  52   meta = with lib; {
  53     description = "Python bindings for the Message Passing Interface standard";
  54     homepage = "https://bitbucket.org/mpi4py/mpi4py/";
  55     license = licenses.bsd3;
  56   };
  57 }