Dominik Roth
46f742c1ee
daf8ec2 Making strating_pos configurable git-subtree-dir: subtrees/columbus git-subtree-split: daf8ec2c4dc75dac8c9a005d882ef681b02df19f
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,
|
|
)
|