Merge pull request #69 from Maximellerbach/new-circuit

New circuit
tawnkramer · Jul 25, 2021 · 484fb8a · 484fb8a
2 parents 78fa6be + 02102ed
commit 484fb8a
Show file tree

Hide file tree

Showing 9 changed files with 15 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -119,6 +119,7 @@ Example info:
 - "donkey-minimonaco-track-v0"
 - "donkey-warren-track-v0"
 - "donkey-thunderhill-track-v0"
+- "donkey-circuit-launch-track-v0"
 
 
 ## Codestyle

diff --git a/examples/genetic_alg/simple_gen_driver.py b/examples/genetic_alg/simple_gen_driver.py
@@ -192,6 +192,7 @@ def on_agents_finished(self):
         "donkey-minimonaco-track-v0",
         "donkey-warren-track-v0",
         "donkey-thunderhill-track-v0",
+        "donkey-circuit-launch-track-v0",
     ]
 
     parser = argparse.ArgumentParser(description="simple_gen_driver")

diff --git a/examples/gym_test.py b/examples/gym_test.py
@@ -71,6 +71,7 @@ def exit_scene(env):
         "donkey-minimonaco-track-v0",
         "donkey-warren-track-v0",
         "donkey-thunderhill-track-v0",
+        "donkey-circuit-launch-track-v0",
     ]
 
     parser = argparse.ArgumentParser(description="gym_test")

diff --git a/examples/reinforcement_learning/ddqn.py b/examples/reinforcement_learning/ddqn.py
@@ -340,6 +340,7 @@ def signal_handler(signal, frame):
         "donkey-minimonaco-track-v0",
         "donkey-warren-track-v0",
         "donkey-thunderhill-track-v0",
+        "donkey-circuit-launch-track-v0",
     ]
 
     parser = argparse.ArgumentParser(description="ddqn")

diff --git a/examples/reinforcement_learning/ppo_train.py b/examples/reinforcement_learning/ppo_train.py
@@ -49,6 +49,7 @@ def _init():
         "donkey-minimonaco-track-v0",
         "donkey-warren-track-v0",
         "donkey-thunderhill-track-v0",
+        "donkey-circuit-launch-track-v0",
     ]
 
     parser = argparse.ArgumentParser(description="ppo_train")

diff --git a/examples/test_cam_config.py b/examples/test_cam_config.py
@@ -20,6 +20,7 @@
         "donkey-minimonaco-track-v0",
         "donkey-warren-track-v0",
         "donkey-thunderhill-track-v0",
+        "donkey-circuit-launch-track-v0",
     ]
 
     parser = argparse.ArgumentParser(description="gym_test")

diff --git a/gym_donkeycar/__init__.py b/gym_donkeycar/__init__.py
@@ -7,6 +7,7 @@
 
 from gym_donkeycar.envs.donkey_env import (
     AvcSparkfunEnv,
+    CircuitLaunchEnv,
     GeneratedRoadsEnv,
     GeneratedTrackEnv,
     MiniMonacoEnv,
@@ -46,3 +47,5 @@
 register(id="donkey-warren-track-v0", entry_point="gym_donkeycar.envs.donkey_env:WarrenTrackEnv")
 
 register(id="donkey-thunderhill-track-v0", entry_point="gym_donkeycar.envs.donkey_env:ThunderhillTrackEnv")
+
+register(id="donkey-circuit-launch-track-v0", entry_point="gym_donkeycar.envs.donkey_env:CircuitLaunchEnv")
diff --git a/gym_donkeycar/envs/donkey_env.py b/gym_donkeycar/envs/donkey_env.py
@@ -191,3 +191,8 @@ def __init__(self, *args, **kwargs):
 class ThunderhillTrackEnv(DonkeyEnv):
     def __init__(self, *args, **kwargs):
         super(ThunderhillTrackEnv, self).__init__(level="thunderhill", *args, **kwargs)
+
+
+class CircuitLaunchEnv(DonkeyEnv):
+    def __init__(self, *args, **kwargs):
+        super(CircuitLaunchEnv, self).__init__(level="circuit_launch", *args, **kwargs)
diff --git a/tests/test_gym_donkeycar.py b/tests/test_gym_donkeycar.py
@@ -15,6 +15,7 @@
     "donkey-minimonaco-track-v0",
     "donkey-warren-track-v0",
     "donkey-thunderhill-track-v0",
+    "donkey-circuit-launch-track-v0",
 ]