From df4361564adfe80be77eb8f812d7bc8780097cbe Mon Sep 17 00:00:00 2001
From: ottofabian <fabi.otto@t-online.de>
Date: Fri, 4 Sep 2020 13:35:05 +0200
Subject: [PATCH] added max entropy RL and loading/testing of models

---
 alr_envs/__init__.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/alr_envs/__init__.py b/alr_envs/__init__.py
index 7f50314..571bc42 100644
--- a/alr_envs/__init__.py
+++ b/alr_envs/__init__.py
@@ -11,6 +11,15 @@ register(
     entry_point='alr_envs.classic_control:SimpleReacherEnv',
     max_episode_steps=200,
     kwargs={
-        "n_links": 3,
+        "n_links": 2,
+    }
+)
+
+register(
+    id='SimpleReacher5-v0',
+    entry_point='alr_envs.classic_control:SimpleReacherEnv',
+    max_episode_steps=200,
+    kwargs={
+        "n_links": 5,
     }
 )