def _decompose(example: dict) -> list[dict]:
        """
        Decompose the example into a list of dict instances.

        Args:
            example: a dict instance that has multiple keys
        """
        num_chunks = len(example[list(example.keys())[0]])
        return [{k: example[k][i] for k in example.keys()} for i in range(num_chunks)]

        Decompose the example into a list of dict instances.

        Args:
            example: a dict instance that has multiple keys
        
smoe/data/aggregation.py