From b7b5af03bafea82e5c52ed5048ae224640bbdc03 Mon Sep 17 00:00:00 2001
From: ottofabian <fabi.otto@t-online.de>
Date: Mon, 22 Feb 2021 17:05:12 +0100
Subject: [PATCH] added kl full cov projection and cov plots

---
 alr_envs/mujoco/alr_reacher.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/alr_envs/mujoco/alr_reacher.py b/alr_envs/mujoco/alr_reacher.py
index d72c9ce..5b13203 100644
--- a/alr_envs/mujoco/alr_reacher.py
+++ b/alr_envs/mujoco/alr_reacher.py
@@ -29,7 +29,7 @@ class ALRReacherEnv(mujoco_env.MujocoEnv, utils.EzPickle):
         else:
             raise ValueError(f"Invalid number of links {n_links}, only 5 or 7 allowed.")
 
-        utils.EzPickle.__init__(**locals())
+        utils.EzPickle.__init__(steps_before_reward=steps_before_reward, n_links=n_links, balance=balance)
         mujoco_env.MujocoEnv.__init__(self, os.path.join(os.path.dirname(__file__), "assets", file_name), 2)
 
     def step(self, a):