Test framework for BabyRobotEnv_v2

BabyRobotEnv_v2_Runner.py

# create the environment
env = gym.make('BabyRobotEnv-v2')

# initialize the environment
env.reset()
env.render()

terminated = False
while not terminated:  

  # choose a random action
  action = env.action_space.sample()   

  # take the action and get the information from the environment
  new_state, reward, terminated, truncated, info = env.step(action)
  
  # show the current position and reward
  env.render(action=action, reward=reward)