def init_sim()

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

• 17 lines of code
• 4 McCabe index (conditional complexity)

    def init_sim(self, root: mjcf.RootElement, frameskip: int = 5):
        W = 8
        add_fwd_corridor(root, W)
        # 200 obstacles should be enough for everybody
        self.n_obstacles = 200
        for i in range(self.n_obstacles):
            if i % 2 == 0:
                b = add_box(
                    root, f'hurdle-{i}', size=[0.05, W, 0.1], pos=[2, 0, 0.2]
                )
            else:
                b = add_capsule(
                    root,
                    f'bar-{i}',
                    fromto=[2.025, -W, 0.1, 2.025, W, 0.1],
                    size=[0.1],
                )

        super().init_sim(root, frameskip)