Fixed bug with value_func-renderer

2022-10-16 19:15:23 +02:00 · 2022-10-16 19:15:23 +02:00 · ba796a1d6e
commit ba796a1d6e
parent e371d871e6
1 changed files with 1 additions and 1 deletions
--- a/columbus/env.py
+++ b/columbus/env.py
@ -323,7 +323,7 @@ class ColumbusEnv(gym.Env):
            c = 0
            for i in range(resolution):
                for j in range(resolution):
-                    v = V[c]
+                    v = V[c].item()
                    c += 1
                    col = [int((1-v)*color_depth),
                           int(v*color_depth), 0, color_depth]