Fixed bug with value_func-renderer

This commit is contained in:
2022-10-16 19:15:23 +02:00
parent e371d871e6
commit ba796a1d6e
+1 -1
View File
@@ -323,7 +323,7 @@ class ColumbusEnv(gym.Env):
c = 0 c = 0
for i in range(resolution): for i in range(resolution):
for j in range(resolution): for j in range(resolution):
v = V[c] v = V[c].item()
c += 1 c += 1
col = [int((1-v)*color_depth), col = [int((1-v)*color_depth),
int(v*color_depth), 0, color_depth] int(v*color_depth), 0, color_depth]