WUT_Computer_Science/script/main.py

10 lines
319 B
Python
Raw Normal View History

2024-10-27 21:14:26 +01:00
from metadrive.envs.metadrive_env import MetaDriveEnv
import gymnasium as gym
env = MetaDriveEnv(config={"use_render": True})
obs, info = env.reset()
for i in range(1000):
obs, reward, terminated, truncated, info = env.step(env.action_space.sample())
if terminated or truncated:
env.reset()
env.close()