'The implementation of this method should have two parts, structured
like the following:
<perform calculations before stepping the world>
yield
reward = <perform calculations after stepping the world>
yield reward'

def compute_reward(self, action):

    raise NotImplementedError
