chore: revert configs and reset tests

instadeepai · Jan 25, 2024 · e9c4fe1 · e9c4fe1
1 parent ca11e71
commit e9c4fe1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/jumanji/environments/routing/pac_man/env_test.py b/jumanji/environments/routing/pac_man/env_test.py
@@ -78,7 +78,8 @@ def test_pac_man__step_jit(pac_man: PacMan) -> None:
     # Check that the state has changed
     assert new_state.player_locations.x == state.player_locations.x
     assert new_state.player_locations.y != state.player_locations.y
-    assert not jnp.array_equal(new_state.ghost_locations, state.ghost_locations)
+    # Ghosts should not move on turn 1
+    assert jnp.array_equal(new_state.ghost_locations, state.ghost_locations)
     assert new_state.pellets != state.pellets
 
 

diff --git a/jumanji/training/configs/config.yaml b/jumanji/training/configs/config.yaml
@@ -1,10 +1,10 @@
 defaults:
     - _self_
-    - env: pac_man  # [snake, cleaner, connector, cvrp, game_2048, graph_coloring, job_shop, knapsack, maze, minesweeper, mmst, multi_cvrp, robot_warehouse, rubiks_cube, snake, sudoku, tetris, tsp]
+    - env: snake  # [snake, cleaner, connector, cvrp, game_2048, graph_coloring, job_shop, knapsack, maze, minesweeper, mmst, multi_cvrp, robot_warehouse, rubiks_cube, snake, sudoku, tetris, tsp]
 
 agent: random  # [random, a2c]
 
-seed: 12
+seed: 0
 
 logger:
     type: neptune  # [neptune, tensorboard, terminal]