Fixed bug with value_func-renderer
This commit is contained in:
parent
e371d871e6
commit
ba796a1d6e
@ -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]
|
||||||
|
Loading…
Reference in New Issue
Block a user