def select_actions(self):
    assert (self.policy_array is not None)
    probs = self.policy_array[self.state.get_state_idx(self.policy_idx_type)]
    aev_idx = np.random.choice(np.arange(Action.NUM_ACTIONS_TOTAL), p=probs)
    return Action(action_idx=aev_idx)
def select_actions(self):
    assert (self.policy_array is not None)
    probs = self.policy_array[self.state.get_state_idx(self.policy_idx_type)]
    aev_idx = np.random.choice(np.arange(Action.NUM_ACTIONS_TOTAL), p=probs)
    return Action(action_idx=aev_idx)