mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-07-11 23:42:40 +02:00
started visualization of routes in plot single runs, assets missing.
This commit is contained in:
12
test_run.py
12
test_run.py
@ -1,4 +1,5 @@
|
||||
from pathlib import Path
|
||||
from pprint import pprint
|
||||
|
||||
from tqdm import trange
|
||||
|
||||
@ -7,12 +8,17 @@ from marl_factory_grid.algorithms.static.TSP_item_agent import TSPItemAgent
|
||||
from marl_factory_grid.algorithms.static.TSP_target_agent import TSPTargetAgent
|
||||
from marl_factory_grid.environment.factory import Factory
|
||||
|
||||
from marl_factory_grid.utils.plotting.plot_single_runs import plot_routes
|
||||
|
||||
if __name__ == '__main__':
|
||||
# Render at each step?
|
||||
|
||||
run_path = Path('study_out')
|
||||
render = True
|
||||
monitor = True
|
||||
record = 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)
|
||||
@ -34,3 +40,5 @@ if __name__ == '__main__':
|
||||
if done:
|
||||
print(f'Episode {episode} done...')
|
||||
break
|
||||
|
||||
plot_routes(factory, agents, )
|
||||
|
Reference in New Issue
Block a user