def test_plaintext_to_pretokenized_rename(self):
    ds = self.cached_plaintext_task.get_dataset({'inputs': 13, 'targets': 13}, split='train', use_cached=True, shuffle=False)
    keys = next(ds.as_numpy_iterator()).keys()
    self.assertSetEqual(set(keys), set(['inputs', 'inputs_pretokenized', 'targets', 'targets_pretokenized']))
def test_plaintext_to_pretokenized_rename(self):
    ds = self.cached_plaintext_task.get_dataset({'inputs': 13, 'targets': 13}, split='train', use_cached=True, shuffle=False)
    keys = next(ds.as_numpy_iterator()).keys()
    self.assertSetEqual(set(keys), set(['inputs', 'inputs_pretokenized', 'targets', 'targets_pretokenized']))