&`i TdZddlZddlmZddlmZmZmZmZddl Z ddl Z ddl m Z ddlmZedZGddZeGd d Zd Zd Ze ddedeededeedeeeeff dZdeeeeffdZdS)z-A utility for debugging serialization issues.N)contextmanager)AnyOptionalSetTuple) DeveloperAPIc#RK|xjdz c_dV|xjdzc_dS)N)level)printers l/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/util/check_serialize.py_indentrs3 MMQMM EEE MMQMMMMc eZdZdZdZdZdS)_Printerc"d|_||_dS)Nr)r print_file)selfrs r __init__z_Printer.__init__s $rc t|SN)rrs r indentz_Printer.indentst}}rcLd|jz}t||z|jdS)Nz )file)r printr)rmsgrs r rz_Printer.prints-$*$ fsl111111rN)__name__ __module__ __qualname__rrrrr rrsA%%%22222rrc,eZdZdZdededefdZdZdS) FailureTuplezRepresents the serialization 'frame'. Attributes: obj: The object that fails serialization. name: The variable name of the object. parent: The object that references the `obj`. objnameparentc0||_||_||_dSr)r$r%r&)rr$r%r&s r rzFailureTuple.__init__,s  rc8d|jd|jd|jdS)Nz FailTuple(z [obj=z , parent=z]))r%r$r&rs r __repr__zFailureTuple.__repr__1s)ODIOOTXOO OOOOrN)rrr __doc__rstrrr)r!rr r#r#"s[CsC PPPPPrr#c tj|sJtj|}d}|jr|dt |jd|5|jD])\}}t|||dz |||\} } |p| }|rn*dddn #1swxYwY|j r|dt |j d|5|j D])\}}t|||dz |||\} } |p| }|rn*dddn #1swxYwY|s|d|d |S) AAdds the first-found non-serializable element to the failure_set.Fz Detected z. global variables. Checking serializability...r r%depthr& failure_setr Nz0 nonlocal variables. Checking serializability...1WARNING: Did not find non-serializable object in . This may be an oversight.) inspect isfunctiongetclosurevarsglobalsrlenritems_inspect_serializability nonlocals) base_objr/r&r0r closurefoundr%r$ serializable_s r _inspect_func_serializationr@5s  h ' ''' '$X..G E  *GO,, * * *   ^^    $_2244   c":!)! +# ### a1\!1E                 *G-.. * * *   ^^    $.4466   c":!)! +# ### a1\!1E                   ( ( ( (    Ls&3ACC CAE$$E(+E(c tj|rJtj|tj}d}|5|D])\}}t |||dz |||\} } |p| }|rn*dddn #1swxYwY|5tj|} | D]h\}}|dr|dstj|rDt |||dz |||\} } |p| }|rnidddn #1swxYwY|s|d|d|S) r-) predicateFr r.N__r1r2) r3r4 getmembersrr9 startswithendswith isbuiltinr) r;r/r&r0r functionsr=r%r$r>r?memberss r _inspect_generic_serializationrJhsP!(++++ +"8w7IJJJI E     "  ID#6ai' OL!--E                    $X..  ID#t$$ t)<)< @QRU@V@V 6ai' OL!--E      ( ( ( (    Ls% -BB  B %BD11D58D5r;r%r/rreturncHt|}t|||dd|S)aIdentifies what objects are preventing serialization. Args: base_obj: Object to be serialized. name: Optional name of string. depth: Depth of the scope stack to walk through. Defaults to 3. print_file: file argument that will be passed to print(). Returns: bool: True if serializable. set[FailureTuple]: Set of unserializable objects. .. versionadded:: 1.1.0 N)rr9)r;r%r/rr s r inspect_serializabilityrNs),z""G #HdE4w O OOrctjd}d}d}|d}t}d|}|dt t |dz|||dt t |dz|t |}n|d|d|d  tj|d|fS#t$r} |tj j d tj j d | d} |d kr$| t|||n#t$rYnwxYwYd} ~ nd} ~ wwxYw|d krd|fStj|rt#||||| nt%||||| |s$| t||||rR|dt t |dz|s|dnwdtjjdd|Dztjjdz} |d| d|d|dt t |dz|d|d|dt t |dz| |fS)NFTzChecking Serializability of =Pz Serializing 'z' z...z!!! FAILz serialization: r)r/r&r0r zTNothing failed the inspect_serialization test, though serialization did not succeed.z  c34K|]}t|VdSr)r+).0ks r z+_inspect_serializability..s(88qCFF888888rz z Variable: zowas found to be non-serializable. There may be multiple other undetected variables that were non-serializable. zConsider either removing the instantiation/imports of these variables or moving the instantiation into the scope of the function/class. zmCheck https://docs.ray.io/en/master/ray-core/objects/serialization.html#troubleshooting for more information.zIf you have any suggestions on how to improve this error message, please reach out to the Ray developers on github.com/ray-project/ray/issues/)coloramainitsetrminr7r+cpdumps ExceptionForeREDRESETaddr#r3r4r@rJStyleBRIGHTjoin RESET_ALL) r;r%r/r&r0r top_level declarationr=e fail_varss r r9r9s MOOOIK E ee ?X??  cCK 0 0"555666 k""" cCK 0 0"555666 <x==D ;d;;h;;;<<<  [      } V V(-*= V VST V V    zz XtV D DEEE    D   zzk!! (## # #       ' #      > XtV<<=== 7 cCK 0 0"555666  MM1     1.00))88K888889^-3334  MM%Y%%%    MMG     cCK 0 0"555666 {      C    cCK 0 0"5556669k !!s<C33 F==E;;*E&%E;& E30E;2E33E;;F)NrKN)r*r3 contextlibrtypingrrrrrXrayray.cloudpickle cloudpickler\ray.util.annotationsrrrr#r@rJr+intboolrNr9r!rr rss33%%%%%%,,,,,,,,,,,, ------ 2 2 2 2 2 2 2 2PPPPPPPP$000f(((V $ PPP 3-P P P  4\" "# PPPP2]" 4\" "#]"]"]"]"]"]"r