pass seed during env.reset

Moribund7 · Nov 5, 2017 · 4f0c537 · 4f0c537
1 parent 31eaece
commit 4f0c537
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/osim/redis/service.py b/osim/redis/service.py
@@ -82,7 +82,7 @@ def run(self):
                         return self._error_template(_error_message)
                     else:
                         self.env = RunEnv(visualize = _payload['visualize'], max_obstacles=10)
-                        _observation = self.env.reset()
+                        _observation = self.env.reset(seed=self.seed_map[self.simulation_count]))
                         _observation = np.array(_observation).tolist()
                         _command_response = {}
                         _command_response['type'] = messages.OSIM_RL.ENV_CREATE_RESPONSE
@@ -99,7 +99,7 @@ def run(self):
                     """
                     self.simulation_count += 1
                     if self.seed_map and self.simulation_count < len(self.seed_map):
-                        _observation = self.env.reset()
+                        _observation = self.env.reset(seed=self.seed_map[self.simulation_count])
                         _observation = list(_observation)
                         _command_response = {}
                         _command_response['type'] = messages.OSIM_RL.ENV_RESET_RESPONSE