def reset_state()

in bisk/tasks/limbo.py [0:0]

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

    def reset_state(self):
        super().reset_state()
        self.max_bars_cleared = 0
        xpos = 1
        intervals = self.np_random.uniform(3, 6, size=(self.n_bars,))
        if self.fixed_height:
            heights = np.zeros(self.n_bars) + self.min_height
        else:
            heights = self.np_random.uniform(
                self.min_height, self.min_height + 0.3, size=(self.n_bars,)
            )
        self.bar_pos = []
        nm = self.p.named.model
        for i in range(self.n_bars):
            xpos += intervals[i]
            self.bar_pos.append(xpos)
            nm.geom_pos[f'bar-{i}'][0] = xpos
            nm.geom_pos[f'bar-{i}'][2] = (
                heights[i] + nm.geom_size[f'bar-{i}'][0]
            )
            nm.geom_rgba[f'bar-{i}'] = [0.8, 0.9, 0.8, 1]
        self.bar_hit = [False] * self.n_bars
        self.new_bars_hit = set()