Learning Representations in Video Game Agents with Supervised Contrastive Imitation Learning