updated stuff

2021-05-07 14:45:04 +02:00
parent 020288fd55
commit 83d399901b
2 changed files with 26 additions and 4 deletions
--- a/environments/helpers.py
+++ b/environments/helpers.py
@@ -16,7 +16,7 @@ def parse_level(path):

 def one_hot_level(level, wall_char=WALL):
    grid = np.array(level)
-    binary_grid = np.zeros(grid.shape)
+    binary_grid = np.zeros(grid.shape, dtype=np.int8)
    binary_grid[grid == wall_char] = 1
    return binary_grid