def get_legal_actions(self, agent_index=0): """ Returns the legal actions for the agent specified (0 is red_bird) (State, int) -> [Action] """ if self.terminal: return [] if agent_index == 0: from game_rules import RedBirdRules return RedBirdRules.get_legal_actions(self) else: from game_rules import BlackBirdRules return BlackBirdRules.get_legal_actions(self)