U Nv f@slddlZddlmZddlmZmZiZGdddZGdddeZdd Z d d Z d d Z dddZ dS)N)partial)dumpsloadsc@s&eZdZd ddZddZddZdS) CloudpickledObjectWrapperFcCs||_||_dSN_obj _keep_wrapper)selfobj keep_wrapperr M/tmp/pip-unpacked-wheel-dylwa62s/joblib/externals/loky/cloudpickle_wrapper.py__init__ sz"CloudpickledObjectWrapper.__init__cCs(t|j}|jst|ffSt||jffSr)rrr r_reconstruct_wrapper)r _pickled_objectr r r __reduce__s  z$CloudpickledObjectWrapper.__reduce__cCs|dkrt|j|St||S)Nr)getattrr)r attrr r r __getattr__s z%CloudpickledObjectWrapper.__getattr__N)F)__name__ __module__ __qualname__rrrr r r rr s rc@seZdZddZdS)CallableObjectWrappercOs |j||Sr)rr argskwargsr r r__call__szCallableObjectWrapper.__call__N)rrrrr r r rrsrcCs t|rt||dSt||dS)Nr )callablerr)r r r r r_wrap_non_picklable_objects#s r cCst|}t||Sr)rr )rr r r r rr)srcCsdt|ddk}t|trLtt|jfdd|jDdd|jDSt|rt|dd}|t|d d t j @O}t|d d}|d |kO}|s|St |}|dkrt |d d}|t |<|S)N__main__rcSsg|] }t|qSr _wrap_objects_when_needed).0ar r r 5sz-_wrap_objects_when_needed..cSsi|]\}}|t|qSr r#)r%kvr r r 6sz-_wrap_objects_when_needed..__code__co_flagsrrzFr)r isinstancerr$funcrkeywordsitemsrinspect CO_NESTED WRAP_CACHEgetr )r Z need_wrap func_code func_nameZ wrapped_objr r rr$.s.      r$Tcs8tr,Gfdddt}j|_|StdS)aWrapper for non-picklable object to use cloudpickle to serialize them. Note that this wrapper tends to slow down the serialization process as it is done with cloudpickle which is typically slower compared to pickle. The proper way to solve serialization issues is to avoid defining functions and objects in the main scripts and to implement __reduce__ functions for complex classes. cseZdZfddZdS)z.CloudpickledClassWrappercs|||_|_dSrrrr r r rr]s zEwrap_non_picklable_objects..CloudpickledClassWrapper.__init__N)rrrrr r7r rCloudpickledClassWrapper\sr8r)r1isclassrrr )r r r8r r7rwrap_non_picklable_objectsOs r:)T) r1 functoolsrZjoblib.externals.cloudpicklerrr3rrr rr$r:r r r rs !