dd2eaf6c174fdd757514bc06d1063e5f9c6bb4a0,torchnlp/samplers/noisy_sorted_sampler.py,NoisySortedSampler,__iter__,#NoisySortedSampler#,44

Before Change

        self.sorted_indexes = [item[0] for item in zip_]

    def __iter__(self):
        return iter(self.sorted_indexes)

    def __len__(self):
        return len(self.data)

After Change

        self.get_noise = get_noise

    def __iter__(self):
        zip_ = []
        for i, row in enumerate(self.data):
            value = self.get_noise(row) + self.sort_key(row)
            zip_.append(tuple([i, value]))
        zip_ = sorted(zip_, key=lambda r: r[1])
        return iter([item[0] for item in zip_])

    def __len__(self):
        return len(self.data)

Instances