Get an action using this trainer's current policy.
        :param curr_info: Current BrainInfo.
        :return: The ActionInfo given by the policy given the BrainInfo.

def get_action(self, curr_info: BrainInfo) -> ActionInfo:
        """
        Get an action using this trainer's current policy.
        :param curr_info: Current BrainInfo.
        :return: The ActionInfo given by the policy given the BrainInfo.
        """
        self.trainer_metrics.start_experience_collection_timer()
        action = self.policy.get_action(curr_info)
        self.trainer_metrics.end_experience_collection_timer()
        return action