code: def choose_action(self, curr_state):
      self.current_high_level_control = curr_state[
          self.ap.algorithm.state_key_with_the_class_index
      ]
      return super().choose_action(curr_state)
  
imports: from typing import Union
  from rl_coach.agents.imitation_agent import ImitationAgent
  from rl_coach.architectures.embedder_parameters import InputEmbedderParameters
  from rl_coach.architectures.head_parameters import RegressionHeadParameters
  from rl_coach.architectures.middleware_parameters import FCMiddlewareParameters
  from rl_coach.base_parameters import AgentParameters, MiddlewareScheme, NetworkParameters, AlgorithmParameters
  from rl_coach.exploration_policies.e_greedy import EGreedyParameters
  from rl_coach.memories.non_episodic.balanced_experience_replay import BalancedExperienceReplayParameters
label_desc: Use image node_id 2 for calling the CILAgent obj's underlying member method code with example usage: obj.choose_action(curr_state) and returns: super