hongbo-miao/hongbomiao.com

import gymnasium as gym


def main() -> None:
    env = gym.make("CartPole-v1")
    observation, info = env.reset(seed=42)
    print(observation, info)
    for _ in range(1000):
        action = env.action_space.sample()
        observation, reward, terminated, truncated, info = env.step(action)

        if terminated or truncated:
            observation, info = env.reset()
            print(observation, info)
    env.close()


if __name__ == "__main__":
    main()