def test_num_epochs(self):
    epoch1_ds = self.random_task.get_dataset({'inputs': 13, 'targets': 13}, split='train', use_cached=False, shuffle=True, seed=0)
    epoch2_ds = self.random_task.get_dataset({'inputs': 13, 'targets': 13}, split='train', use_cached=False, shuffle=True, seed=0).repeat(2).skip(3)
    test_utils.assert_datasets_eq(epoch1_ds, epoch2_ds)
    epoch1_ds = self.random_task.get_dataset({'inputs': 13, 'targets': 13}, split='train', use_cached=False, shuffle=True, seed=0)
    epoch2_ds = self.random_task.get_dataset({'inputs': 13, 'targets': 13}, split='train', use_cached=False, shuffle=True, seed=0, num_epochs=2).skip(3)
    test_utils.assert_datasets_neq(epoch1_ds, epoch2_ds)
def test_num_epochs(self):
    epoch1_ds = self.random_task.get_dataset({'inputs': 13, 'targets': 13}, split='train', use_cached=False, shuffle=True, seed=0)
    epoch2_ds = self.random_task.get_dataset({'inputs': 13, 'targets': 13}, split='train', use_cached=False, shuffle=True, seed=0).repeat(2).skip(3)
    test_utils.assert_datasets_eq(epoch1_ds, epoch2_ds)
    epoch1_ds = self.random_task.get_dataset({'inputs': 13, 'targets': 13}, split='train', use_cached=False, shuffle=True, seed=0)
    epoch2_ds = self.random_task.get_dataset({'inputs': 13, 'targets': 13}, split='train', use_cached=False, shuffle=True, seed=0, num_epochs=2).skip(3)
    test_utils.assert_datasets_neq(epoch1_ds, epoch2_ds)