dodox/metastable-baselines
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed reference to gym.spaces

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-06-25 17:49:46 +02:00
parent 5f7cfd2e10
commit df21e1dc3f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
sb3_trl/trl_pg/trl_pg.py
View File

@ -439,7 +439,7 @@ class TRL_PG(OnPolicyAlgorithm):
# Rescale and perform action # Rescale and perform action
clipped_actions = actions clipped_actions = actions
# Clip the actions to avoid out of bound error # Clip the actions to avoid out of bound error
if isinstance(self.action_space, gym.spaces.Box): if isinstance(self.action_space, spaces.Box):
clipped_actions = np.clip( clipped_actions = np.clip(
actions, self.action_space.low, self.action_space.high) actions, self.action_space.low, self.action_space.high)
@ -455,7 +455,7 @@ class TRL_PG(OnPolicyAlgorithm):
self._update_info_buffer(infos) self._update_info_buffer(infos)
n_steps += 1 n_steps += 1
if isinstance(self.action_space, gym.spaces.Discrete): if isinstance(self.action_space, spaces.Discrete):
# Reshape in case of discrete action # Reshape in case of discrete action
actions = actions.reshape(-1, 1) actions = actions.reshape(-1, 1)