From 253ce6495e9de08cb14f99ecfc1ee20e7d49fbd5 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sun, 19 Jun 2022 22:58:49 +0200 Subject: [PATCH] Added no_map feature to RayTracingEnv --- columbus/env.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/columbus/env.py b/columbus/env.py index a771342..978e29f 100644 --- 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