Use cloudpickle when executing Fiber pool in interactive sessions

uber · Jul 8, 2020 · fb4bac5 · fb4bac5
1 parent b17769c
commit fb4bac5
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/fiber/pool.py b/fiber/pool.py
@@ -35,7 +35,6 @@
 import multiprocessing as mp
 import multiprocessing.pool as mp_pool
 import multiprocessing.util as mp_util
-import pickle
 import math
 import queue
 import random
@@ -57,6 +56,12 @@
 from fiber.process import current_process
 import signal
 
+
+if fiber.util.is_in_interactive_console():
+    import cloudpickle as pickle
+else:
+    import pickle
+
 logger = logging.getLogger('fiber')
 
 MIN_PORT = 40000