def main()

in gym3/viewer.py [0:0]

• 8 lines of code
• 2 McCabe index (conditional complexity)

def main():
    import procgen

    env = procgen.ProcgenGym3Env(num=1, env_name="coinrun", render_mode="rgb_array")
    env = ViewerWrapper(env=env, info_key="rgb")
    start = time.time()
    for i in range(10000):
        env.act(types_np.sample(env.ac_space, bshape=(env.num,)))
        print("step", i, i / (time.time() - start))