dodox/fancy_gym
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

minor bug fixes

Browse Source
This commit is contained in:
Fabian
2022-07-25 11:18:05 +02:00
parent 5590318329
commit ce795669a9
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fancy_gym/black_box/black_box_wrapper.py
+3 -3
View File
@@ -158,12 +158,12 @@ class BlackBoxWrapper(gym.ObservationWrapper):
if self.verbose >= 2:
infos['positions'] = trajectory
infos['velocities'] = velocity
infos['step_actions'] = actions[:t + 1]
infos['step_actions'] = actions[:t]
infos['step_observations'] = observations[:t + 1]
infos['step_rewards'] = rewards[:t + 1]
infos['step_rewards'] = rewards[:t]
infos['trajectory_length'] = t + 1
trajectory_return = self.reward_aggregation(rewards[:t + 1])
trajectory_return = self.reward_aggregation(rewards[:t])
return self.observation(obs), trajectory_return, done, infos
def render(self, **kwargs):