steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-07-05 09:01:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

fixed globalpos and 2rooms1door destination spawning

Browse Source
This commit is contained in:
Chanumask
2024-03-25 12:30:31 +01:00
parent ce7f5cb1ea
commit 955598a129
4 changed files with 22 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test_run.py
View File

@ -9,10 +9,10 @@ from marl_factory_grid.environment.factory import Factory
if __name__ == '__main__':
# Render at each step?
render = False
render = True
# Path to config File
path = Path('marl_factory_grid/configs/test_config.yaml')
path = Path('marl_factory_grid/configs/two_rooms_one_door.yaml')
# Env Init
factory = Factory(path)
@ -23,7 +23,8 @@ if __name__ == '__main__':
if render:
factory.render()
action_spaces = factory.action_space
agents = [TSPDirtAgent(factory, 0), TSPItemAgent(factory, 1), TSPTargetAgent(factory, 2)]
# agents = [TSPDirtAgent(factory, 0), TSPItemAgent(factory, 1), TSPTargetAgent(factory, 2)]
agents = [TSPTargetAgent(factory, 0), TSPTargetAgent(factory, 1)]
while not done:
a = [x.predict() for x in agents]
obs_type, _, _, done, info = factory.step(a)