use int instead of math.floor; fixes #32

spro · Jun 8, 2017 · 60a930f · 60a930f
1 parent 5ad67e5
commit 60a930f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/reinforce-gridworld/reinforce-gridworld.py b/reinforce-gridworld/reinforce-gridworld.py
@@ -115,7 +115,7 @@ def reset(self):
         """Start a new episode by resetting grid and agent"""
         self.grid.reset()
         self.agent.reset()
-        c = math.floor(self.grid.grid_size / 2)
+        c = int(self.grid.grid_size / 2)
         self.agent.pos = (c, c)
 
         self.t = 0