o i@sdZddlZddlZddlZddlZddlZddlmZddlm Z ddZ iZ ddZ d d Z d d ZejZGd ddZddZddZeeeddZddZddZddZddZGdddejZdd Zeeje<Gd!d"d"ejZGd#d$d$ZdS)%z/ Serialization support for compiled functions. N) cloudpickle)ircGs |j|S)zM Global hook to rebuild a given class from its __reduce__ arguments. )_rebuild)clsargsrQ/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/numba/core/serialize.py_rebuild_reductions r cCs>||f}zt|}W|Styt|t|<}Y|Sw)zUsed by `numba_unpickle` from _helperlib.c Parameters ---------- address : int bytedata : bytes hashed : bytes Returns ------- obj : object unpickled object )_unpickled_memoKeyErrorrloads)addressbytedatahashedkeyobjrrr_numba_unpickles  rcCsPt}t}||dd}|||}Wd|S1s!wY|S)zISimilar to `pickle.dumps()`. Returns the serialized object in bytes. )protocolN) NumbaPicklerioBytesIOdumpgetvalue)rpicklerbufppickledrrrdumps4s     rcCsZt|\}}}g}t|}|D]}t|tjr |t|q||q|t||fSN) rr iter isinstancerValueappendnexttuple) static_excexc_argsexc static_argslocinfo real_args exc_args_iterargrrrruntime_build_excinfo_struct@s  r.c@s0eZdZdZdZddZddZeddZd S) _CustomPickledaA wrapper for objects that must be pickled with `NumbaPickler`. Standard `pickle` will pick up the implementation registered via `copyreg`. This will spawn a `NumbaPickler` instance to serialize the data. `NumbaPickler` overrides the handling of this type so as not to spawn a new pickler for the object when it is already being pickled by a `NumbaPickler`. ctorstatescCs||_||_dSrr0)selfr1r2rrr__init__]s z_CustomPickled.__init__cCstj|j|jffSr)r/rr1r2r3rrr_reduceaz_CustomPickled._reducecCs |||Srr)rr1r2rrrrds z_CustomPickled._rebuildN) __name__ __module__ __qualname____doc__ __slots__r4r6 classmethodrrrrrr/Ps r/cCst|\}}t||S)zPstandard unpickling for `_CustomPickled`. Uses `NumbaPickler` to load. )r r/) serializedr1r2rrr_unpickle__CustomPickledis  r?cCst|j|jf}t|ffS)zNstandard pickling for `_CustomPickled`. Uses `NumbaPickler` to dump. )rr1r2r?)cpr>rrr_pickle__CustomPickledrs rAcCstt||ffS)aFor customizing object serialization in `__reduce__`. Object states provided here are used as keyword arguments to the `._rebuild()` class method. Parameters ---------- states : dict Dictionary of object states to be serialized. Returns ------- result : tuple This tuple conforms to the return type requirement for `__reduce__`. )custom_rebuildr/)rr2rrr custom_reducesrCcCs|j|j}}|jdi|S)ziCustomized object deserialization. This function is referenced internally by `custom_reduce()`. Nr)r1r2r)custom_pickledrr2rrrrBsrBc Csjt'}t|}z||Wntjy!YWddSw WddS1s.wYdS)zCheck if *obj* can be serialized. Parameters ---------- obj : object Returns -------- can_serialize : bool NFT)rrrrpickle PicklingError)rfoutrrrris_serialiables $rHcCstdt|d)Nz Pickling of z is unsupported)rErFtype)rrrr _no_picklesrJcCstj||S)z1This is called on a type to disable pickling )rdisabled_typesadd)typrrrdisable_picklings rNcs$eZdZeZ fddZZS)rcs"t||jvr t|t|Sr)rIrKrJsuperreducer_override)r3r __class__rrrPs zNumbaPickler.reducer_override)r8r9r:setrKrP __classcell__rrrQrrsrcCs|Sr)r6)r@rrr_custom_reduce__custompickledsrUc@s<eZdZdZejddZejddZddZ dd Z d S) ReduceMixinznA mixin class for objects that should be reduced by the NumbaPickler instead of the standard pickler. cCtrNotImplementedErrorr5rrr_reduce_stateszReduceMixin._reduce_statescKrWrrX)rkwargsrrrrr[zReduceMixin._rebuildcCs|jSrrQr5rrr _reduce_classszReduceMixin._reduce_classcCst||Sr)rCr]rZr5rrr __reduce__r7zReduceMixin.__reduce__N) r8r9r:r;abcabstractmethodrZabstractclassmethodrr]r^rrrrrVs   rVc@s4eZdZdZddZddZddZedd Zd S) PickleCallableByPathaWrap a callable object to be pickled by path to workaround limitation in pickling due to non-pickleable objects in function non-locals. Note: - Do not use this as a decorator. - Wrapped object must be a global that exist in its parent module and it can be imported by `from the_module import the_object`. Usage: >>> def my_fn(x): >>> ... >>> wrapped_fn = PickleCallableByPath(my_fn) >>> # refer to `wrapped_fn` instead of `my_fn` cCs ||_dSr_fn)r3fnrrrr4s zPickleCallableByPath.__init__cOs|j|i|Srrc)r3rr\rrr__call__szPickleCallableByPath.__call__cCst|j|jj|jjffSr)rIrrdr9r8r5rrrr^szPickleCallableByPath.__reduce__cCs|ttj||Sr)getattrsysmodules)rmodnamefn_pathrrrrszPickleCallableByPath._rebuildN) r8r9r:r;r4rfr^r=rrrrrrbsrb)r;rhr_rcopyregrEnumbarllvmliterr r rrr.r r/r?rArCrBrHrJrN CloudPicklerrrUdispatch_tableABCrVrbrrrrs8