Switching between different state representations in reinforcement learning