Added no_map feature to RayTracingEnv

2022-06-19 22:58:49 +02:00 · 2022-06-19 22:58:49 +02:00 · 253ce6495e
commit 253ce6495e
parent 3dc5bed9f4
1 changed files with 2 additions and 1 deletions
--- a/columbus/env.py
+++ b/columbus/env.py
@ -232,7 +232,7 @@ class ColumbusTest3_1(ColumbusEnv):


 class ColumbusTestRay(ColumbusEnv):
-    def __init__(self):
+    def __init__(self, hide_map=False):
        super(ColumbusTestRay, self).__init__(
            observable=observables.RayObservable())
        self.start_pos = [0.6, 0.3]
@ -240,3 +240,4 @@ class ColumbusTestRay(ColumbusEnv):
        self.score = 0
        self.reward_mult = 0.001
        self.aux_reward_max = 1
+        self.draw_entities = not hide_map