Fix ezpickle ant

Farama-Foundation · Mar 5, 2024 · 1450ee9 · 1450ee9
1 parent fae0954
commit 1450ee9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mo_gymnasium/envs/mujoco/ant.py b/mo_gymnasium/envs/mujoco/ant.py
@@ -22,7 +22,7 @@ class MOAntEnv(AntEnv, EzPickle):
 
     def __init__(self, cost_objective=True, **kwargs):
         super().__init__(**kwargs)
-        EzPickle.__init__(self, **kwargs)
+        EzPickle.__init__(self, cost_objective, **kwargs)
         self.cost_objetive = cost_objective
         self.reward_dim = 3 if cost_objective else 2
         self.reward_space = Box(low=-np.inf, high=np.inf, shape=(self.reward_dim,))