when I run the `reward_labelling.py` , it occurs during the reward annotation process. Anyway to fix it?