From c0f331f8f07d671bc5cca54adc8249360d65cbf3 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Sun, 26 Jun 2022 10:56:14 +0200
Subject: [PATCH] New Env: ColumbusCandyland_Aux10

---
 columbus/env.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/columbus/env.py b/columbus/env.py
index 11ea687..79c1eef 100644
--- a/columbus/env.py
+++ b/columbus/env.py
@@ -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,