“A reward function can be compactly specified, similar to a single line of Python code.”