Fixed bug in value_func visualization

2022-10-19 10:41:49 +02:00 · 2022-10-19 10:41:49 +02:00 · d73c7798b3
commit d73c7798b3
parent 89e5f1369e
1 changed files with 1 additions and 2 deletions
--- a/columbus/env.py
+++ b/columbus/env.py
@ -308,8 +308,7 @@ class ColumbusEnv(gym.Env):
            obs = []
            for i in range(resolution):
                for j in range(resolution):
-                    x, y = (i+0.5)*(self.width/resolution), (j+0.5) *\
-                        (self.height/resolution)
+                    x, y = (i+0.5)/resolution, (j+0.5)/resolution
                    self.agent.pos = x, y
                    ob = self.observable.get_observation()
                    obs.append(ob)