Skip to content

Commit fc02030

Browse files
author
Wen-Tse Chen
committed
fix set reward bug
1 parent f8879b3 commit fc02030

File tree

1 file changed

+1
-13
lines changed

1 file changed

+1
-13
lines changed

openrl/envs/nlp/daily_dialog_env.py

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -111,21 +111,9 @@ def __init__(
111111
self.__time_step = None
112112
self.reward_function = None
113113

114-
self.set_reward()
115-
116114
def set_reward(self, reward_fn=None):
117115

118-
from openrl.envs.nlp.rewards.meteor import Meteor
119-
120-
meteor_config = {
121-
"meteor_coeff": 0.5,
122-
"test": False,
123-
}
124-
self.reward_function = {
125-
"meteor": Meteor(**meteor_config),
126-
}
127-
128-
# self.reward_function = reward_fn
116+
self.reward_function = reward_fn
129117

130118
def step_word(self, word: str) -> Tuple[Dict[str, torch.tensor], int, bool, dict]:
131119
action = self.tokenizer.encode(word)[1]

0 commit comments

Comments
 (0)