dodox/Columbus
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

New Env: ColumbusCandyland_Aux10

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-06-26 10:56:14 +02:00
parent 4f3c8bec8b
commit c0f331f8f0
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
columbus/env.py
View File

@ -321,6 +321,12 @@ class ColumbusCandyland(ColumbusEnv):
self.entities.append(reward)
class ColumbusCandyland_Aux10(ColumbusCandyland):
def __init__(self):
super(ColumbusCandyland_Aux10, self).__init__()
self.aux_reward_max = 10
class ColumbusEasyObstacles(ColumbusEnv):
def __init__(self, observable=observables.RayObservable(num_rays=16), hide_map=False, fps=30, env_seed=None):
super(ColumbusEasyObstacles, self).__init__(
@ -445,6 +451,12 @@ register(
max_episode_steps=30*60*2,
)
register(
id='ColumbusCandyland_Aux10-v0',
entry_point=ColumbusCandyland_Aux10,
max_episode_steps=30*60*2,
)
register(
id='ColumbusEasyObstacles-v0',
entry_point=ColumbusEasyObstacles,