if done set pot to 0

fschlatt · Jan 10, 2021 · bc9bac5 · bc9bac5
1 parent 5907506
commit bc9bac5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clubs/poker/engine.py b/clubs/poker/engine.py
@@ -375,8 +375,8 @@ def step(self, bet: float) -> Tuple[Dict, np.ndarray, np.ndarray]:
         observation, payouts, done = self._output()
         if all(done):
             self.action = -1
+            self.pot = 0
             observation["action"] = -1
-            observation["pot"] = 0
         return observation, payouts, done
 
     def _render_config(self):