import torch


def _make_encoder_decoder_data(num_micro_batches=4, micro_batch_size=4, seq_len=64, target_seq_len=64):
    """Make a batch of sequences."""
    return tuple(
        dict(
            input_ids=torch.randint(low=1, high=100, size=(micro_batch_size, seq_len)),
            decoder_input_ids=torch.randint(low=1, high=100, size=(micro_batch_size, target_seq_len)),
        )
        for _ in range(num_micro_batches)
    )