o Xi@sFdZddlmZddlZddlmZddlmZGdddejZ dS)z0Class for storing metadata about the IR objects.) annotationsN)Mapping)AnycsLeZdZdZddfdd Zdd dZdddZdddZdddZZ S) MetadataStorea2Class for storing metadata about the IR objects. Metadata is stored as key-value pairs. The keys are strings and the values can be any Python object. The metadata store also supports marking keys as invalid. This is useful when a pass wants to mark a key that needs to be recomputed. NdataMapping[str, Any] | NonereturnNonecst|t|_dSN)super__init__set _invalid_keys)selfr __class__E/home/ubuntu/.local/lib/python3.10/site-packages/onnx_ir/_metadata.pyr s  zMetadataStore.__init__keystritemrcCs||j|<|j|dSr )rrdiscard)rrrrrr __setitem__s zMetadataStore.__setitem__cCs|j|dSr )raddrrrrr invalidateszMetadataStore.invalidateboolcCs ||jvS)aVReturns whether the value is valid. Note that default values (None) are not necessarily invalid. For example, a shape that is unknown (None) may be still valid if shape inference has determined that the shape is unknown. Whether a value is valid is solely determined by the user that sets the value. )rrrrris_valid!s zMetadataStore.is_validcCs|jjd|jd|jdS)N(z, invalid_keys=))r__name__rr)rrrr__repr__,szMetadataStore.__repr__r )rrrr )rrrrrr )rrrr )rrrr)rr) r __module__ __qualname____doc__r rrrr! __classcell__rrrrr s    r) r$ __future__r collectionscollections.abcrtypingrUserDictrrrrrs