main.py

import argparse

from utils.config import *
from agents import *


def main():
    # parse the path of the json config file
    arg_parser = argparse.ArgumentParser(description="")
    arg_parser.add_argument('config',
                            metavar='config',
                            default='None',
                            help='The Configuration file in json format')
    args = arg_parser.parse_args()

    config, _ = get_config_from_json(args.config)
    if config.multi_agent:
        multi_params  = config[config.multi_param]
        for i in multi_params:
            config[config.multi_param] = i
            config.exp_name = os.path.join(config.multi_exp_name, 
                                            'exp_' + str(i))
            config = process_config(config)
            run_agent(config)
    else:
        config = process_config(config)
        run_agent(config)

def run_agent(config):
    agent_class = globals()[config.agent]
    agent = agent_class(config)
    agent.run()
    agent.finalize()

if __name__ == '__main__':
    main()