mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-12-10 17:40:37 +01:00
Adapted commit: "started visualization of routes in plot single runs, assets missing."
This commit is contained in:
committed by
Julian Schönberger
parent
4571dc1cd1
commit
0d5b20a16f
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)
|
||||
@@ -33,3 +39,5 @@ if __name__ == '__main__':
|
||||
if done:
|
||||
print(f'Episode {episode} done...')
|
||||
break
|
||||
|
||||
plot_routes(factory, agents, )
|
||||
|
||||
Reference in New Issue
Block a user