added extra check

MSU-AI · Mar 12, 2024 · dea1078 · dea1078
1 parent ea61435
commit dea1078
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/clash_royale/envs/game_engine/game_engine.py b/clash_royale/envs/game_engine/game_engine.py
@@ -169,7 +169,7 @@ def terminal_value(self) -> int:
         """
         Returns side won, otherwise returns -1.
         """
-        
+
         player1_val: int = self.arena.tower_count(0)
         player2_val: int = self.arena.tower_count(1)
         if player1_val == player2_val:

diff --git a/clash_royale/envs/game_engine/player.py b/clash_royale/envs/game_engine/player.py
@@ -57,7 +57,7 @@ def step(self,
              elixir_rate: float,
              frames: int=1) -> None:
         """
-        Called with the value of elixir_rate and frame to update the elixir of player after 'frame' number of frames
+        Called with the value of elixir_rate and frame to update the elixir of player
         to better customize the elixir_rate that can vary depends on game modes.
         """
 
@@ -72,7 +72,8 @@ def pop(self, card_index: int) -> None:
 
         """
 
-        assert(card_index < 4)
+        assert card_index >= 0 and card_index < 4
+
         self.deck.put(self.hand[card_index], block = False)
         self.hand[card_index] = self.next
         self.next: Card = self.deck.get(block = False)