@@ -46,6 +46,7 @@ class RecordEpisodeStatistics(VectorWrapper):
4646 Example:
4747 >>> from pprint import pprint
4848 >>> import gymnasium as gym
49+ >>> from gymnasium.wrappers.vector import RecordEpisodeStatistics
4950 >>> envs = gym.make_vec("CartPole-v1", num_envs=3)
5051 >>> envs = RecordEpisodeStatistics(envs)
5152 >>> obs, info = envs.reset(123)
@@ -58,14 +59,9 @@ class RecordEpisodeStatistics(VectorWrapper):
5859 >>> envs.close()
5960 >>> pprint(info) # doctest: +SKIP
6061 {'_episode': array([ True, False, False]),
61- '_final_info': array([ True, False, False]),
62- '_final_observation': array([ True, False, False]),
6362 'episode': {'l': array([11, 0, 0], dtype=int32),
6463 'r': array([11., 0., 0.], dtype=float32),
6564 't': array([0.007812, 0. , 0. ], dtype=float32)},
66- 'final_info': array([{}, None, None], dtype=object),
67- 'final_observation': array([array([ 0.11448676, 0.9416149 , -0.20946532, -1.7619033 ], dtype=float32),
68- None, None], dtype=object)}
6965 """
7066
7167 def __init__ (
0 commit comments