def scan_recorded_traces()

in gym_recording/playback.py [0:0]

• 9 lines of code
• 5 McCabe index (conditional complexity)

def scan_recorded_traces(directory, episode_cb=None, max_episodes=None):
    """
    Go through all the traces recorded to directory, and call episode_cb for every episode.
    Set max_episodes to end after a certain number (or you can just throw an exception from episode_cb
    if you want to end the iteration early)
    """
    rdr = TraceRecordingReader(directory)
    added_episode_count = 0
    for batch in rdr.get_recorded_batches():
        for ep in rdr.get_recorded_episodes(batch):
            episode_cb(ep['observations'], ep['actions'], ep['rewards'])
            added_episode_count += 1
            if max_episodes is not None and added_episode_count >= max_episodes: return
    rdr.close()