From 741f1cb6360c55846c407e9286c89a7b0a27036b Mon Sep 17 00:00:00 2001
From: ottofabian <fabi.otto@t-online.de>
Date: Tue, 3 Nov 2020 11:26:06 +0100
Subject: [PATCH] smaller changes

---
 alr_envs/__init__.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/alr_envs/__init__.py b/alr_envs/__init__.py
index 2439822..8f0264c 100644
--- a/alr_envs/__init__.py
+++ b/alr_envs/__init__.py
@@ -1,5 +1,7 @@
 from gym.envs.registration import register
 
+from alr_envs.stochastic_search.functions.f_rosenbrock import Rosenbrock
+
 register(
     id='ALRReacher-v0',
     entry_point='alr_envs.mujoco:ALRReacherEnv',
@@ -107,3 +109,24 @@ register(
         "n_links": 5,
     }
 )
+
+
+register(
+    id='SimpleReacher5-v0',
+    entry_point='alr_envs.classic_control:SimpleReacherEnv',
+    max_episode_steps=200,
+    kwargs={
+        "n_links": 5,
+    }
+)
+
+for dim in [5, 10, 25, 50, 100]:
+
+    register(
+        id=f'Rosenbrock{dim}-v0',
+        entry_point='alr_envs.stochastic_search:StochasticSearchEnv',
+        max_episode_steps=1,
+        kwargs={
+            "cost_f": Rosenbrock,
+        }
+    )
\ No newline at end of file