safety

2022-05-29 11:58:01 +02:00
parent 647f086a8d
commit f1a96c055b
12 changed files with 333 additions and 168 deletions
@@ -205,7 +205,7 @@ class EpisodicWrapper(gym.Env, ABC):
            infos['step_actions'] = actions[:t + 1]
            infos['step_observations'] = observations[:t + 1]
            infos['step_rewards'] = rewards[:t + 1]
-            infos['trajectory_length'] = t + 1
+        infos['trajectory_length'] = t + 1
        done = True
        return self.get_observation_from_step(obs), trajectory_return, done, infos