From e9c4fe1e925c8cc30fd3e653ff3aaf6b1c484999 Mon Sep 17 00:00:00 2001 From: siddarthsingh1 Date: Thu, 25 Jan 2024 15:13:02 +0200 Subject: [PATCH] chore: revert configs and reset tests --- jumanji/environments/routing/pac_man/env_test.py | 3 ++- jumanji/training/configs/config.yaml | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/jumanji/environments/routing/pac_man/env_test.py b/jumanji/environments/routing/pac_man/env_test.py index eeee9f55b..f2ab5a7ec 100644 --- 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 index 32269964f..9897e5fd4 100644 --- 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]