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