def demo_run()

in agent_zoo/RoboschoolInvertedPendulumSwingup_v0_2017may.py [0:0]

• 24 lines of code
• 7 McCabe index (conditional complexity)

def demo_run():
    env = gym.make("RoboschoolInvertedPendulumSwingup-v1")
    pi = SmallReactivePolicy(env.observation_space, env.action_space)

    while 1:
        frame = 0
        score = 0
        restart_delay = 0
        obs = env.reset()

        while 1:
            a = pi.act(obs)
            obs, r, done, _ = env.step(a)
            score += r
            frame += 1
            still_open = env.render("human")
            if still_open==False:
                return
            if not done: continue
            if restart_delay==0:
                print("score=%0.2f in %i frames" % (score, frame))
                restart_delay = 60*2  # 2 sec at 60 fps
            else:
                restart_delay -= 1
                if restart_delay > 0: continue
                break