gym.env.render()

rendered_image = env.render(mode="rgb_array")
plt.imshow(rendered_image)

import mediapy as media

# Function to simulate an episode with a random policy
def run_episode(env, frames_per_episode):
    frames = []  # Collect frames for video rendering
    env.reset()
    for _ in range(frames_per_episode):
        frames.append(env.render(mode="rgb_array"))
        action = env.action_space.sample()  # Random action
        new_obs, reward, done, info = env.step(action)
        if done:
            break
    return frames

# Visualize one episode with a random policy
frames = run_episode(env, frames_per_episode)

media.show_video(frames, fps=30)

Dark screen issue