Text this: Inverse reinforcement learning in contextual MDPs.