88a96639ea79b9678fd70d5e5919d5ef7fcb9a89,dataset/pipeline.py,Pipeline,gen_batch,#Pipeline#Any#Any#Any#Any#Any#,266

Before Change

                    is_end = True
        else:
            for batch in batch_generator:
                yield self._exec_all_actions(batch)

        self.reset_iter()

    def next_batch(self, batch_size, shuffle=False, n_epochs=1, drop_last=False, prefetch=0, *args, **kwargs):

After Change

        else:
            for batch in batch_generator:
                try:
                    batch_res = self._exec_all_actions(batch)
                except SkipBatchException:
                    pass
                else:
                    yield batch_res

Instances