Fixes #2

activatedgeek · Feb 17, 2019 · 5370571 · 5370571
1 parent c4322dc
commit 5370571
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gym_2048/env.py b/gym_2048/env.py
@@ -84,6 +84,8 @@ def reset(self):
     self.board = np.zeros((self.width, self.height), dtype=np.int64)
     self._place_random_tiles(self.board, count=2)
 
+    return self.board
+
   def render(self, mode='human'):
     if mode == 'human':
       for row in self.board.tolist():