def _chunk(examples: dict, block_size: int) -> list[dict]:
        """
        Split the concatenated examples into chunks of block_size.

        Args:
            examples: a dict instance that has multiple keys
            block_size: the length of the concatenated examples
        """
        total_length = len(examples[list(examples.keys())[0]])
        if total_length >= block_size:
            total_length = (total_length // block_size) * block_size
        result = {
            k: [t[i : i + block_size] for i in range(0, total_length, block_size)]
            for k, t in examples.items()
        }
        return result

        Split the concatenated examples into chunks of block_size.

        Args:
            examples: a dict instance that has multiple keys
            block_size: the length of the concatenated examples
        
smoe/data/aggregation.py