(env, actions, seed)
| 62 | |
| 63 | |
| 64 | def _rollout_frames(env, actions, seed): |
| 65 | obs, _ = env.reset(seed=seed) |
| 66 | frames = [] |
| 67 | f = _image_from_obs(env, obs) |
| 68 | if f is not None: |
| 69 | frames.append(f) |
| 70 | for a in actions: |
| 71 | out = env.step(a) |
| 72 | obs = out[0] |
| 73 | f = _image_from_obs(env, obs) |
| 74 | if f is not None: |
| 75 | frames.append(f) |
| 76 | return frames |
| 77 | |
| 78 | |
| 79 | def _save_mp4(frames, path, fps=10): |
no test coverage detected