Move importing MPI where it is needed in controller

mila-iqia · Feb 21, 2017 · 54c8a0d · 54c8a0d
1 parent 73b63f6
commit 54c8a0d
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/platoon/channel/controller.py b/platoon/channel/controller.py
@@ -37,11 +37,6 @@
 import posix_ipc
 import zmq
 
-try:
-    from mpi4py import MPI
-except ImportError:
-    MPI = None
-
 from ..util import (PlatoonError, mmap, launch_process,
                     op_to_mpi, dtype_to_mpi)
 
@@ -345,6 +340,10 @@ def _init_region_comm(self):
         .. versionadded:: 0.6.0
 
         """
+        try:
+            from mpi4py import MPI
+        except ImportError:
+            MPI = None
         if MPI is None:
             raise AttributeError("mpi4py is not imported")
         self._region_comm = MPI.COMM_WORLD