_cloudpickle_wrapper.py 416 B

123456789101112131415161718
  1. """
  2. Small shim of loky's cloudpickle_wrapper to avoid failure when
  3. multiprocessing is not available.
  4. """
  5. from ._multiprocessing_helpers import mp
  6. def _my_wrap_non_picklable_objects(obj, keep_wrapper=True):
  7. return obj
  8. if mp is not None:
  9. from .externals.loky import wrap_non_picklable_objects
  10. else:
  11. wrap_non_picklable_objects = _my_wrap_non_picklable_objects
  12. __all__ = ["wrap_non_picklable_objects"]