o $i\ @sddlZddlmZddlmZddlmZddlmZ m Z ddl m Z ddl Z z ddlmZmZWneyCdd Zd d ZYnwd d ZdS)N) OrderedDict)Iterator)getitem)getquote)apply)fields is_dataclasscCsdS)NFxr r Q/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/util/dask/common.pyr r cCsgSNr r r r r dataclass_fieldsrrcs`gitjfddtjtfdd|Df<fdd}|fS)a Extract Ray object refs from a set of potentially arbitrarily nested Python objects. Intended use is to find all Ray object references in a set of (possibly nested) Python objects, do something to them (get(), wait(), etc.), then repackage them into equivalent Python objects. Args: *args: One or more (potentially nested) Python objects that contain Ray object references. Returns: A 2-tuple of a flat list of all contained Ray object references, and a function that, when given the corresponding flat list of concrete values, will return a set of Python objects equivalent to that which was given in *args, but with all Ray object references replaced with their corresponding concrete values. csttjr}ttf|<|Stj}tt r&t nt }|t t t fvr=|fddDf}n-|ttfvrQ|fddDf}ntrht|dtfddtDff}nS||<|S)Ncg|]}|qSr r .0i_unpackr r :z7unpack_object_refs.._unpack..cs g|] \}}||gqSr r )rkvrr r r<s r cs"g|] }|jt|jgqSr )namegetattr)rf)rexprr r rDs) isinstanceray ObjectRefhexrlenappenduuiduuid4rlisttypetuplesetdictritemsr rr)rtokentyp repack_task)r object_refsobject_refs_token repack_dsk)rr r/s0      z#unpack_object_refs.._unpackcrr r rrr r rPrz&unpack_object_refs..cs}t||<t|Sr)copyrget_sync)resultsdsk)r1outr2r r repackRs  z"unpack_object_refs..repack)r%r&r"r))argsr8r )rr0r1r7r2r unpack_object_refss  r:)r% collectionsrcollections.abcroperatorr dask.corerr4r dask.utilsrr dataclassesrrr ImportErrorr:r r r r s