allow for non-mpi tests

2019-01-24 14:35:41 -08:00
parent 547764efc9
commit e868bdaa1a
1 changed files with 4 additions and 1 deletions
--- a/baselines/common/tests/test_with_mpi.py
+++ b/baselines/common/tests/test_with_mpi.py
@@ -5,7 +5,10 @@ import cloudpickle
 import base64
 import pytest

-from mpi4py import MPI
+try:
+    from mpi4py import MPI
+except ImportError:
+    MPI = None

 def test_with_mpi(nproc=2, timeout=30, skip_if_no_mpi=True):
    def outer_thunk(fn):