fix set reward bug

OpenRL-Lab · Dec 20, 2023 · fc02030 · fc02030
1 parent f8879b3
commit fc02030
Showing 1 changed file with 1 addition and 13 deletions.
diff --git a/openrl/envs/nlp/daily_dialog_env.py b/openrl/envs/nlp/daily_dialog_env.py
@@ -111,21 +111,9 @@ def __init__(
         self.__time_step = None
         self.reward_function = None
 
-        self.set_reward()
-
     def set_reward(self, reward_fn=None):
 
-        from openrl.envs.nlp.rewards.meteor import Meteor
-
-        meteor_config = {
-            "meteor_coeff": 0.5,
-            "test": False,
-        }
-        self.reward_function = {
-            "meteor": Meteor(**meteor_config),
-        }
-
-        # self.reward_function = reward_fn
+        self.reward_function = reward_fn
 
     def step_word(self, word: str) -> Tuple[Dict[str, torch.tensor], int, bool, dict]:
         action = self.tokenizer.encode(word)[1]