mirror of
https://github.com/kuhyx/WUT_Computer_Science.git
synced 2026-07-04 16:03:11 +02:00
11 lines
435 B
Python
11 lines
435 B
Python
from metadrive.envs.metadrive_env import MetaDriveEnv
|
|
import gymnasium as gym
|
|
from metadrive.envs.gym_wrapper import createGymWrapper # import the wrapper
|
|
|
|
env = createGymWrapper(MetaDriveEnv)(config={"use_render": True}) # wrap the environment
|
|
obs = env.reset()
|
|
for i in range(1000):
|
|
obs, reward, done, info = env.step(env.action_space.sample()) # the return value contains no truncate
|
|
if done:
|
|
env.reset()
|
|
env.close() |