From ba796a1d6e88d632147986405addd14d058a50b1 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sun, 16 Oct 2022 19:15:23 +0200 Subject: [PATCH] Fixed bug with value_func-renderer --- columbus/env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/columbus/env.py b/columbus/env.py index 2b6d02d..dc4b17d 100644 --- 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]