updated stuff

This commit is contained in:
romue
2021-05-07 14:45:04 +02:00
parent 020288fd55
commit 83d399901b
2 changed files with 26 additions and 4 deletions

View File

@ -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