Function _selected_state_indices

tools/render/render_episode.py:1252–1263

(
    *,
    state_count: int,
    stride: int,
    max_frames: int,
)

def _selected_state_indices(
    *,
    state_count: int,
    stride: int,
    max_frames: int,
) -> list[int]:
    if state_count <= 0:
        return []
    indices = list(range(0, state_count, max(1, int(stride))))
    if int(max_frames) > 0:
        indices = indices[: int(max_frames)]
    return indices


def _camera_rgb(camera: Any):