76bedc874bc399c4171086d7fce4121e3fc8f860,deeppavlov/data/dataset.py,Dataset,batch_generator,#Dataset#Any#Any#,16

Before Change

        Returns:
            batch_gen: a generator, that iterates through the part (defined by data_type) of dataset
        
        raise NotImplementedError("Batch generator must be implemented for Dataset class!")

    def iter_all(self):
        rIterate through all data. It can be used for building dictionary or

After Change

            batch_gen: a generator, that iterates through the part (defined by data_type) of dataset
        
        data = self.data[data_type]
        data_len = len(data)
        order = list(range(data_len))
        random.shuffle(order)
        for i in range((data_len - 1) // batch_size + 1):
            yield list(zip(*[data[o] for o in order[i*batch_size:(i+1)*batch_size]]))

    def iter_all(self, data_type: str = "train") -> Generator:
        rIterate through all data. It can be used for building dictionary or

        Returns:

Instances