Files
zenml/venv/lib/python3.9/site-packages/joblib/_cloudpickle_wrapper.py
Christian Mantha 2ca0b9ef7c star
2026-03-02 19:10:52 -05:00

19 lines
416 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 not None:
from .externals.loky import wrap_non_picklable_objects
else:
wrap_non_picklable_objects = _my_wrap_non_picklable_objects
__all__ = ["wrap_non_picklable_objects"]