Skip to content

Commit 91dea6d

Browse files
committed
update
1 parent 28c7f07 commit 91dea6d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

verl/workers/config/reward_model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff 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.

0 commit comments

Comments
 (0)