看好自己代码中import的highway_env包在什么位置再进行修改
pip show highway_env
1.在envs文件夹中仿照其他环境创建自己的环境文件myHighway.py,需要修改类名,例如:
仿照highway_env0.py时,需要将
class HighwayEnv(AbstractEnv):
修改为自己的环境名:
class MyHighwayEnv(AbstractEnv):
2.添加注册信息
在envs文件夹中的__init__.py末尾添加:
register(
id='myHighway-v0',#调用环境时的id,需要以"-v"加数字为结尾,否则报错
entry_point='highway_env.envs:MyHighway',
)
在envs文件夹中的__init__.py开头添加:
from gym import register
from highway_env.envs.myHighway import *
3.测试环境
import gym
import pprint
from highway_env.envs.common.abstract import AbstractEnv#需要在建立环境的代码中加入此行
env_name = 'myHighway-v0'
env = gym.make(env_name)