def __next__()

in sagemaker/source/dataset/dataset_generator.py [0:0]

• 10 lines of code
• 3 McCabe index (conditional complexity)

    def __next__(self):
        state = self.state
        delta = np.random.normal(0, self.std*self.std_factor)
        if self.state > self.mean + self.cap:
            self.state -= abs(delta)
        elif self.state < self.mean - self.cap:
            self.state += abs(delta)
        else:
            self.state += delta

        return state