highway-env环境搭建及测试代码

#import gym
import gymnasium as gym
import highway_env
from matplotlib import pyplot as plt

env = gym.make('intersection-v0',render_mode="rgb_array")
env.reset()

for _ in range(100):
    action = env.action_type.actions_indexes["IDLE"]
    obs, reward, down, info,_ = env.step(action)
    env.render()
screen = env.render()
plt.imshow(env.render(screen))
plt.show()

gym==0.26.2