20 lines
634 B
Python
20 lines
634 B
Python
import os
|
|
import gymnasium as gym
|
|
from gym_donkeycar.envs.donkey_env import DonkeyEnv
|
|
from autoresearch.envs.gymnasium import GymnasiumEnv
|
|
|
|
# Choose your desired track:
|
|
TRACK = os.environ.get("DONKEYCAR_TRACK", "generated_road") # You can override with env var
|
|
|
|
print(f"[AutoResearch] Using DonkeyCar track: {TRACK}")
|
|
donkey_env = DonkeyEnv(level=TRACK)
|
|
|
|
# Wrap it as an autoresearch env
|
|
aresearch_env = GymnasiumEnv(donkey_env)
|
|
|
|
# Import and run an agent from autoresearch
|
|
from autoresearch.agents.dqn import main as dqn_main
|
|
# Run for a few episodes with default settings
|
|
if __name__ == '__main__':
|
|
dqn_main(env=aresearch_env)
|