when-to-switch

1
import os
2

3
from prettytable import PrettyTable
4

5
from agents.assistant_switcher import example_assistant_switcher
6
from agents.epom import example_epom
7
from agents.heuristic_switcher import example_heuristic_switcher
8
from agents.learnable_switcher import example_learnable_switcher
9
from agents.replan import example_replan
10

11

12
def example(map_name='sc1-AcrosstheCape', num_agents=64, seed=0, animate=True):
13
    os.environ['OMP_NUM_THREADS'] = "1"
14
    os.environ['MKL_NUM_THREADS'] = "1"
15

16
    run_examples_funcs = [
17
        example_epom,
18
        example_replan,
19
        example_assistant_switcher,
20
        example_heuristic_switcher,
21
        example_learnable_switcher
22
    ]
23

24
    score_table = PrettyTable()
25
    score_table.field_names = ["Algorithm", "ISR", "CSR", "Episode Length"]
26

27
    for run_example_func in run_examples_funcs:
28
        result = run_example_func(map_name=map_name, num_agents=num_agents, seed=seed, animate=animate)
29

30
        if result:
31
            score_table.add_row([result['algorithm'], result['ISR'], result['CSR'], result['ep_length']])
32

33
            print(score_table.get_string(start=len(score_table._rows) - 1, end=len(score_table._rows)))
34

35
    print(score_table)
36

37

38
if __name__ == '__main__':
39
    example()
40