def _name_cleaner(agent_name):
  """Renames agent_name to prettier string for plots."""
  rename_dict = {'correct_ts': 'Correct TS',
                 'kl_ucb': 'KL UCB',
                 'misspecified_ts': 'Misspecified TS',
                 'ucb1': 'UCB1',
                 'ucb-best': 'UCB-best',
                 'nonstationary_ts': 'Nonstationary TS',
                 'stationary_ts': 'Stationary TS',
                 'greedy': 'greedy',
                 'ts': 'TS',
                 'action_0': 'Action 0',
                 'action_1': 'Action 1',
                 'action_2': 'Action 2',
                 'bootstrap': 'bootstrap TS',
                 'laplace': 'Laplace TS',
                 'thoughtful': 'Thoughtful TS',
                 'gibbs': 'Gibbs TS'}
  if agent_name in rename_dict:
    return rename_dict[agent_name]
  else:
    return agent_name