File size: 269 Bytes
e085e3b
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
import gym
env = gym.make("procgen:procgen-starpilot-v0")

obs = env.reset()
step = 0
while True:
    obs, rew, done, info = env.step(env.action_space.sample())
    print(info)
    print(f"step {step} reward {rew} done {done}")
    step += 1
    if done:
        break