14 lines
455 B
Python
14 lines
455 B
Python
|
from gym.envs.registration import register
|
||
|
|
from env import *
|
||
|
|
|
||
|
|
def register():
|
||
|
|
register(
|
||
|
|
# unique identifier for the env `name-version`
|
||
|
|
id="Columbus-Test317-v0",
|
||
|
|
# path to the class for creating the env
|
||
|
|
# Note: entry_point also accept a class as input (and not only a string)
|
||
|
|
entry_point=env.ColumbusEnv,
|
||
|
|
# Max number of steps per episode, using a `TimeLimitWrapper`
|
||
|
|
max_episode_steps=500,
|
||
|
|
)
|