22 lines
675 B
Python
22 lines
675 B
Python
import gymnasium as gym
|
|
import gym_donkeycar.envs
|
|
import gym_donkeycar.envs.donkey_env
|
|
import os
|
|
|
|
host = os.environ.get('DONKEY_SIM_HOST', '10.0.0.55')
|
|
port = int(os.environ.get('DONKEY_SIM_PORT', '9091'))
|
|
|
|
print(f"Attempting to connect to donkey sim at {host}:{port}")
|
|
|
|
# Try gymnasium-style make
|
|
env = gym.make("donkey-generated-roads-v0")
|
|
|
|
observation, info = env.reset()
|
|
for t in range(10):
|
|
action = env.action_space.sample()
|
|
observation, reward, terminated, truncated, info = env.step(action)
|
|
print(f"Step {t}: reward {reward}, done {terminated or truncated}, info {info}")
|
|
if terminated or truncated:
|
|
observation, info = env.reset()
|
|
env.close()
|