diff --git a/README.rst b/README.rst index 92c4ce9da..d207e2633 100644 --- 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 index 7cee6c1dd..cf50f174d 100644 --- 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 index 656232345..64055d9bd 100755 --- 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)