fix: four rooms now constructs goal correctly

epignatelli · Jun 6, 2024 · a42bbb3 · a42bbb3
1 parent 2cf6cc9
commit a42bbb3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/navix/environments/four_rooms.py b/navix/environments/four_rooms.py
@@ -79,7 +79,7 @@ def _reset(self, key: Array, cache: Union[RenderingCache, None] = None) -> Times
             pocket=EMPTY_POCKET_ID,
         )
         # goal
-        goal = Goal(position=goal_pos, probability=jnp.asarray(1.0))
+        goal = Goal.create(position=goal_pos, probability=jnp.asarray(1.0))
 
         entities = {
             Entities.PLAYER: player[None],
@@ -112,7 +112,7 @@ def _reset(self, key: Array, cache: Union[RenderingCache, None] = None) -> Times
         width=19,
         observation_fn=kwargs.pop("observation_fn", observations.symbolic),
         reward_fn=kwargs.pop("reward_fn", rewards.on_goal_reached),
-        termination_fn=kwargs.pop("termination_fn", terminations.on_goal_reached),        
+        termination_fn=kwargs.pop("termination_fn", terminations.on_goal_reached),
         *args,
         **kwargs,
     ),