def METHOD_NAME(
    self,
    states: List[torch.Tensor],
    saveDir: Optional[str] = None,
    frameSkip: Optional[int] = 5,
):
    images = []
    for state in states[::frameSkip]:
        self.setState(state)
        image = self.camera.renderState()
        images.append(image)
    if saveDir is not None:
        self.camera.saveImages(images, saveDir)
    return images