o bi@s<ddlmZddlmZmZmZmZGdddeeZdS))Generic)Block BlockAccessor BlockTypeTc@seZdZdZededdfddZdeddfdd Zdeddfd d Z de fd d Z defddZ de fddZde fddZdefddZdS) BlockBuilderzA builder class for blocks.blockreturncCst|S)N)r for_blockbuilder)rr T/home/ubuntu/.local/lib/python3.10/site-packages/ray/data/_internal/block_builder.pyr szBlockBuilder.for_blockitemNcCt)z-Append a single row to the block being built.NotImplementedError)selfrr r r add zBlockBuilder.addcCr)z0Append an entire block to the block being built.r)rrr r r add_blockrzBlockBuilder.add_blockcCr)z8Whether building this block will yield a new block copy.rrr r r will_build_yield_copyrz"BlockBuilder.will_build_yield_copycCr)zBuild the block.rrr r r buildrzBlockBuilder.buildcCr)z-Return the number of rows added in the block.rrr r r num_rowsrzBlockBuilder.num_rowscCr)z2Return the estimated memory usage so far in bytes.rrr r r get_estimated_memory_usage!rz'BlockBuilder.get_estimated_memory_usagecCr)zReturn the block type.rrr r r block_type%rzBlockBuilder.block_type)__name__ __module__ __qualname____doc__ staticmethodrr rrrboolrrintrrrrr r r r rsrN)typingrray.data.blockrrrrrr r r r s