Traceback (most recent call last):
File "/home/zhsx/Directional-Stimulus-Prompting/scripts/training/train_text_generation.py", line 84, in
main(
File "/home/zhsx/Directional-Stimulus-Prompting/scripts/training/train_text_generation.py", line 46, in main
trainer = OnPolicyTrainer(
File "/home/zhsx/Directional-Stimulus-Prompting/rl4lms/envs/text_generation/training_utils.py", line 149, in init
self._setup()
File "/home/zhsx/Directional-Stimulus-Prompting/rl4lms/envs/text_generation/training_utils.py", line 157, in _setup
self._reward_fn = build_reward_fn(self._reward_config)
File "/home/zhsx/Directional-Stimulus-Prompting/rl4lms/envs/text_generation/training_utils.py", line 47, in build_reward_fn
reward_fn = RewardFunctionRegistry.get(reward_config["id"],
File "/home/zhsx/Directional-Stimulus-Prompting/rl4lms/envs/text_generation/registry.py", line 140, in get
reward_cls = cls._registry[reward_fn_id]
KeyError: 'summarization_with_hint'