首先我们建立下面的文件结构:
gym-foo/ (文件夹)
readme.txt
setup.py
gym_foo/ (文件夹)
__init__.py
envs/ (文件夹)
__init.py__.py
foo_env.py
readme就是个描述,描述这个环境。
gym-foo/setup.py
from setuptools import setup
setup(name='gym_foo',
version='0.0.1',
install_requires=['gym']#And any other dependencies required
)
这里的name变量,我们将用作导入我们的环境(eg. import gym_foo).
gym-foo/gym_foo/__init__.py
from gym.envs.registration import register
register(
id='foo-v0',
entry_point='gym_foo.envs:FooEnv',
)
id变量是我们要放到gym.make()里去调用我们的环境。
gy