Fix last action calculations

hellomyheart · Jun 13, 2023 · e3dd938 · e3dd938
1 parent 98587da
commit e3dd938
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rlcard/envs/doudizhu.py b/rlcard/envs/doudizhu.py
@@ -60,6 +60,7 @@ def _extract_state(self, state):
             for i, action in reversed(state['trace']):
                 if i == 0:
                     last_landlord_action = action
+                    break
             last_landlord_action = _cards2array(last_landlord_action)
             landlord_num_cards_left = _get_one_hot_array(state['num_cards_left'][0], 20)
 
@@ -69,6 +70,7 @@ def _extract_state(self, state):
             for i, action in reversed(state['trace']):
                 if i == teammate_id:
                     last_teammate_action = action
+                    break
             last_teammate_action = _cards2array(last_teammate_action)
             teammate_num_cards_left = _get_one_hot_array(state['num_cards_left'][teammate_id], 17)
             obs = np.concatenate((current_hand,