Type alias shared by reward-related code. Code adopted from https://github.com/HumanCompatibleAI/imitation.git