Merge pull request #47 from Maximellerbach/Added-roboracingleague-track

added new track
tawnkramer · Sep 25, 2020 · 1687f0d · 1687f0d
2 parents ea92be3 + a3a544d
commit 1687f0d
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/README.rst b/README.rst
@@ -97,6 +97,7 @@ Environments
 * "donkey-generated-roads-v0"
 * "donkey-avc-sparkfun-v0"
 * "donkey-generated-track-v0"
+* "donkey-roboracingleague-track-v0"
 
 
 Credits

diff --git a/gym_donkeycar/__init__.py b/gym_donkeycar/__init__.py
@@ -7,7 +7,7 @@
 __version__ = '1.0.16'
 
 from gym.envs.registration import register
-from .envs.donkey_env import GeneratedRoadsEnv, WarehouseEnv, AvcSparkfunEnv, GeneratedTrackEnv, MountainTrackEnv
+from .envs.donkey_env import GeneratedRoadsEnv, WarehouseEnv, AvcSparkfunEnv, GeneratedTrackEnv, MountainTrackEnv, RoboRacingLeagueTrackEnv
 
 register(
     id='donkey-generated-roads-v0',
@@ -33,3 +33,8 @@
     id='donkey-mountain-track-v0',
     entry_point='gym_donkeycar.envs.donkey_env:MountainTrackEnv',
 )
+
+register(
+    id='donkey-roboracingleague-track-v0',
+    entry_point='gym_donkeycar.envs.donkey_env:RoboRacingLeagueTrackEnv',
+)
diff --git a/gym_donkeycar/envs/donkey_env.py b/gym_donkeycar/envs/donkey_env.py
@@ -162,3 +162,9 @@ class MountainTrackEnv(DonkeyEnv):
 
     def __init__(self, *args, **kwargs):
         super(MountainTrackEnv, self).__init__(level=4, *args, **kwargs)
+
+
+class RoboRacingLeagueTrackEnv(DonkeyEnv):
+
+    def __init__(self, *args, **kwargs):
+        super(RoboRacingLeagueTrackEnv, self).__init__(level=5, *args, **kwargs)