o i@s0ddlmZmZddlZeGdddeZdS))Protocolruntime_checkableNc@sPeZdZdZdededeefddZdeefddZd e de j fd d Z d S) _Checkpointablea Interface for checkpointable objects. Implemented as a protocol, implicit subtyping is supported so subclasses do not need to inherit this explicitly. This is to allow arbitrary objects/tensor subclasses to hook into DCP seamlessly through implementing the interface. fqnobjectreturncCtd)zI Return a list of WriteItems based on object's contents. z6_Checkpointable._create_write_items is not implementedNotImplementedError)selfrrr c/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/torch/distributed/_checkpointable.py__create_write_items__z&_Checkpointable.__create_write_items__cCr)zU Return a list of `ChunkStorageMetadata` based on object's contents. z5_Checkpointable._create_chunk_list is not implementedr )r r r r __create_chunk_list__rz%_Checkpointable.__create_chunk_list__indexcCr)zI Return a 'torch.Tensor' shard based on 'MetadataIndex'. z4_Checkpointable._get_tensor_shard is not implementedr )r rr r r __get_tensor_shard__rz$_Checkpointable.__get_tensor_shard__N) __name__ __module__ __qualname____doc__strrlistrrinttorchTensorrr r r r rs r)typing_extensionsrrrrr r r r s