18 lines
376 B
Python
18 lines
376 B
Python
|
"""
|
||
|
|
Small shim of loky's cloudpickle_wrapper to avoid failure when
|
||
|
|
multiprocessing is not available.
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
from ._multiprocessing_helpers import mp
|
||
|
|
|
||
|
|
|
||
|
|
def my_wrap_non_picklable_objects(obj, keep_wrapper=True):
|
||
|
|
return obj
|
||
|
|
|
||
|
|
|
||
|
|
if mp is None:
|
||
|
|
wrap_non_picklable_objects = my_wrap_non_picklable_objects
|
||
|
|
else:
|
||
|
|
from .externals.loky import wrap_non_picklable_objects # noqa
|