o Y۷i @sdZddlmZddlmZer"ddlmZddlmZddlm Z ddl m Z ddl mZdd l mZGd d d e Zed Gd ddeZedGdddeZdS)zAccess path classes.) annotations) TYPE_CHECKING)Sequence)Object)Any)IntEnum)register_objectc@s(eZdZdZdZdZdZdZdZdZ dS) AccessKindz(Kinds of access steps in an access path.rrN) __name__ __module__ __qualname____doc__ATTR ARRAY_ITEMMAP_ITEM ATTR_MISSINGARRAY_ITEM_MISSINGMAP_ITEM_MISSINGrrI/home/ubuntu/vllm_env/lib/python3.10/site-packages/tvm_ffi/access_path.pyr &sr zffi.reflection.AccessStepc@s4eZdZUdZded<ded<erd dd Zd Sd S) AccessStepzAccess step container.intkindrkeyreturnrcCdSNrselfrrr__ffi_shallow_copy__;zAccessStep.__ffi_shallow_copy__Nrr)rrrr__annotations__rr$rrrrr1s rzffi.reflection.AccessPathcs4eZdZUdZded<ded<ded<erSdEd d ZedFd d ZdGddZ dHddZ dIddZ dJddZ dHddZ dIddZdJddZdKd!d"ZdLd$d%ZdKd&d'ZdMfd)d* ZedFd+d,ZdNd.d/ZdNd0d1ZdOd2d3ZdPd5d6ZdPd7d8ZdQd:d;ZdQdDisallow direct construction; use `AccessPath.root()` instead.zcAccessPath can't be initialized directly. Use AccessPath.root() to create a path to the root object)super__init__ ValueErrorr" __class__rrr>ms zAccessPath.__init__cCstS)zCreate a root access path. Returns ------- AccessPath A path representing the root of an object graph. )r(r,rrrrrootus zAccessPath.rootothercCst|tsdS||S)z*Return whether two access paths are equal.F isinstancer(r;r#rCrrr__eq__s  zAccessPath.__eq__cCst|tsdS|| S)z.Return whether two access paths are not equal.TrDrFrrr__ne__s  zAccessPath.__ne__cC ||S)aPCheck if this access path is a prefix of another access path. Parameters ---------- other The access path to check if it is a prefix of this access path Returns ------- bool True if this access path is a prefix of the other access path, False otherwise )r8rFrrr is_prefix_of zAccessPath.is_prefix_ofattr_keycCrI)aCreate an access path to the attribute of the current object. Parameters ---------- attr_key The key of the attribute to access Returns ------- AccessPath The extended access path )r1r#rLrrrattrrKzAccessPath.attrcCrI)aCreate an access path that indicate an attribute is missing. Parameters ---------- attr_key The key of the attribute to access Returns ------- AccessPath The extended access path )r4rMrrr attr_missingrKzAccessPath.attr_missingindexcCrI)zCreate an access path to the item of the current array. Parameters ---------- index The index of the item to access Returns ------- AccessPath The extended access path )r2r#rPrrr array_itemrKzAccessPath.array_itemcCrI)aCreate an access path that indicate an array item is missing. Parameters ---------- index The index of the item to access Returns ------- AccessPath The extended access path )r5rQrrrarray_item_missingrKzAccessPath.array_item_missingrcCrI)zCreate an access path to the item of the current map. Parameters ---------- key The key of the item to access Returns ------- AccessPath The extended access path )r3r#rrrrmap_itemrKzAccessPath.map_itemcCrI)zCreate an access path that indicate a map item is missing. Parameters ---------- key The key of the item to access Returns ------- AccessPath The extended access path )r6rTrrrmap_item_missingrKzAccessPath.map_item_missingcCs|S)zConvert the access path to a list of access steps. Returns ------- access_steps The list of access steps )r:r"rrrto_stepss zAccessPath.to_stepsr&)rr()r-rrr()r-r0rr()r-rrr()r-rrr()r-r(rr7)rr9)rr<)rCrrr7)rCr(rr7)rLr0rr()rPrrr()rrrr()"rrrrr'rr$ staticmethodr,r/r1r2r3r4r5r6r8r:r;r>rBrGrHrJrNrOrRrSrUrVrWr__hash__ __classcell__rrr@rr(@sB                        r(N)r __future__rtypingrcollections.abcrtvm_ffirrenumrcoreregistryr r rr(rrrrs