diff --git a/abmarl/examples/sim/pacman.py b/abmarl/examples/sim/pacman.py index 096d7ee3..d1f9d0c8 100644 --- a/abmarl/examples/sim/pacman.py +++ b/abmarl/examples/sim/pacman.py @@ -356,7 +356,7 @@ def example_grid(self): 'F', 'F', 'F', 'F', 'B', 'W', '_'], ['_', 'W', 'F', 'W', 'W', 'F', 'W', 'W', 'W', 'F', 'W', 'F', 'W', 'W', 'W', 'F', 'W', 'W', 'F', 'W', '_'], - ['_', 'W', 'F', 'F', 'W', 'F', 'F', 'F', 'F', 'F', 'P', 'F', 'F', 'F', + ['_', 'W', 'F', 'F', 'W', 'F', 'F', 'F', 'F', 'F', 'F', 'F', 'F', 'F', 'F', 'F', 'W', 'F', 'F', 'W', '_'], ['_', 'W', 'W', 'F', 'W', 'F', 'W', 'F', 'W', 'W', 'W', 'W', 'W', 'F', 'W', 'F', 'W', 'F', 'W', 'W', '_'], diff --git a/examples/rllib_pacman.py b/examples/rllib_pacman.py index 8ca50b8f..74b353ef 100644 --- a/examples/rllib_pacman.py +++ b/examples/rllib_pacman.py @@ -8,12 +8,12 @@ object_registry = { - 'P': lambda n: PacmanAgent( - id='pacman', - encoding=1, - view_range=2, - render_color='yellow', - ), + # 'P': lambda n: PacmanAgent( + # id='pacman', + # encoding=1, + # view_range=2, + # render_color='yellow', + # ), 'W': lambda n: WallAgent( id=f'wall_{n}', encoding=2, @@ -32,11 +32,20 @@ render_color='r' ), } +extra_agents = { + 'pacman': PacmanAgent( + id='pacman', + encoding=1, + view_range=2, + render_color='yellow', + ) +} sim = MultiAgentWrapper( AllStepManager( PacmanSim.build_sim_from_array( PacmanSim.example_grid, object_registry, + extra_agents=extra_agents, states={'PositionState', 'OrientationState', 'HealthState'}, observers={'AbsoluteEncodingObserver'}, overlapping={1: {3, 4}, 4: {3, 4}},