def pad_text_batch(data_batch, vocab2int):
    """Pad text with <PAD> so that each text of a batch has the same length"""
    max_text = max([len(text) for text in data_batch])
    return [text + [vocab2int['<PAD>']] * (max_text - len(text)) for text in data_batch]