Merge pull request #11320 from anntzer/update-multiprocess-example

dstansby · web-flow · commit a8bd774bb212 · 2018-05-28T19:28:39.000+01:00
Py3fy multiprocess example.
diff --git a/examples/misc/multiprocess_sgskip.py b/examples/misc/multiprocess_sgskip.py
@@ -9,24 +9,11 @@
 Written by Robert Cimrman
 """
 
+import multiprocessing as mp
 import time
-import numpy as np
-
-from multiprocessing import Process, Pipe
-
-# This example will likely not work with the native OSX backend.
-# Uncomment the following lines to use the qt5 backend instead.
-#
-# import matplotlib
-# matplotlib.use('qt5agg')
-#
-# Alternatively, with Python 3.4+ you may add the line
-#
-# import multiprocessing as mp; mp.set_start_method("forkserver")
-#
-# immediately after the ``if __name__ == "__main__"`` check.
 
 import matplotlib.pyplot as plt
+import numpy as np
 
 # Fixing random state for reproducibility
 np.random.seed(19680801)
@@ -90,13 +77,10 @@ def __call__(self, pipe):
 
 class NBPlot(object):
     def __init__(self):
-        self.plot_pipe, plotter_pipe = Pipe()
+        self.plot_pipe, plotter_pipe = mp.Pipe()
         self.plotter = ProcessPlotter()
-        self.plot_process = Process(
-            target=self.plotter,
-            args=(plotter_pipe,)
-        )
-        self.plot_process.daemon = True
+        self.plot_process = mp.Process(
+            target=self.plotter, args=(plotter_pipe,), daemon=True)
         self.plot_process.start()
 
     def plot(self, finished=False):
@@ -117,4 +101,6 @@ def main():
 
 
 if __name__ == '__main__':
+    if plt.get_backend() == "MacOSX":
+        multiprocessing.set_start_method("forkserver")
     main()
