From 3eb938e0de145ad9aed521dc3789e6a87845670a Mon Sep 17 00:00:00 2001
From: Marcel <unsux@student.kit.edu>
Date: Tue, 3 Aug 2021 16:51:54 +0200
Subject: [PATCH] Introduce short version of the continuous mountain car env

---
 alr_envs/__init__.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/alr_envs/__init__.py b/alr_envs/__init__.py
index 5a92082..e9c09db 100644
--- a/alr_envs/__init__.py
+++ b/alr_envs/__init__.py
@@ -191,6 +191,14 @@ register(
     }
 )
 
+# Short Continuous Mountain Car
+register(
+    id="MountainCarContinuous-v1",
+    entry_point="gym.envs.classic_control:Continuous_MountainCarEnv",
+    max_episode_steps=100,
+    reward_threshold=90.0,
+)
+
 # Motion Primitive Environments
 
 ## Simple Reacher
@@ -817,10 +825,10 @@ register(
 
 ## Open AI
 register(
-    id='ContinuousMountainCarDetPMP-v0',
+    id='ContinuousMountainCarDetPMP-v1',
     entry_point='alr_envs.utils.make_env_helpers:make_detpmp_env_helper',
     kwargs={
-        "name": "gym.envs.classic_control:MountainCarContinuous-v0",
+        "name": "alr_envs:MountainCarContinuous-v1",
         "wrappers": [open_ai.classic_control.continuous_mountain_car.MPWrapper],
         "mp_kwargs": {
             "num_dof": 1,