&`i'ddlZddlmZddlmZmZddlmZmZddl m Z eGddZ eGdd Z Gd d e Z Gd d ejZdS)N) dataclass)AnyList)Block DataBatch) ObjectRefceZdZUdZeed<dS) BatchMetadatazMetadata associated with a batch. Attributes: batch_idx: The global index of this batch so that downstream operations can maintain ordering. batch_idxN)__name__ __module__ __qualname____doc__int__annotations__/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/data/_internal/block_batching/interfaces.pyr r s%NNNNNrr c(eZdZUdZeed<eed<dS)Batchz~A batch of data. Attributes: metadata: Metadata associated with this batch. data: The batch of data. metadatadataN)r r rrr rrrrrrrs1 OOOOOrrceZdZUdZeed<dS) CollatedBatchzA batch of collated data. Attributes: data: The batch of data which is the output of a user provided collate_fn Therefore, the type of this data can be Any. rN)r r rrrrrrrrr"s% IIIIIrrcbeZdZdZejdeeefdZ de fdZ dZ dS)BlockPrefetcherz!Interface for prefetching blocks.blockscdS)z*Prefetch the provided blocks to this node.Nr)selfrs rprefetch_blockszBlockPrefetcher.prefetch_blocks0s  rreturncdS)z@Return the number of blocks currently scheduled for prefetching.rrrs rnum_prefetched_blocksz%BlockPrefetcher.num_prefetched_blocks5sqrcdS)z'Stop prefetching and release resources.Nrr#s rstopzBlockPrefetcher.stop9s rN) r r rrabcabstractmethodrrrr rr$r&rrrrr-sw++ d9U+;&<    s     rr) metaclass)r' dataclassesrtypingrrray.data.blockrr ray.typesrr rrABCMetarrrrr/s/ !!!!!!++++++++            E            r