File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -51,10 +51,10 @@ class RewardManagerConfig(BaseConfig):
5151 def __post_init__ (self ):
5252 super ().__post_init__ ()
5353 if self .source == "register" :
54- from verl .experimental .reward_loop .reward_manager .registry import REWARD_LOOP_MANAGER_REGISTRY
54+ from verl .experimental .reward_loop .reward_manager .registry import REWARD_MANAGER
5555
56- assert self .name in REWARD_LOOP_MANAGER_REGISTRY , (
57- f"Reward manager is not registered: { self .name = } ,{ REWARD_LOOP_MANAGER_REGISTRY .keys ()= } "
56+ assert self .name in REWARD_MANAGER , (
57+ f"Reward manager is not registered: { self .name = } ,{ REWARD_MANAGER .keys ()= } "
5858 )
5959 elif self .source == "importlib" :
6060 # NOTE: The existence is not checked since it depends on which machine the config is initialized on.
You can’t perform that action at this time.
0 commit comments