steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-07-11 23:42:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Moved route caching to env level and removed print statements

Browse Source
This commit is contained in:
Chanumask
2024-04-28 13:52:39 +02:00
parent 54d4e1ecb5
commit 0bbf0dafdb
4 changed files with 84 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test_run.py
View File

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