code

# main.py

if __name__ == '__main__':
    args = parse_args()
    if args.option == 'train':
        train(args)
    else:
        evaluate(args)

 

 

# set env
def init_env(config, port=0):
    # get scenario
    scenario = config.get('scenario')
    if scenario.startswith('atsc'): # atsc env: set port parameter
        if scenario.endswith('large_grid'): # atsc-large_grid env
            return LargeGridEnv(config, port=port)
        else: # atsc-real_net env
            return RealNetEnv(config, port=port)
    else: # cacc env
        return CACCEnv(config)

 

 

# large_grid_env.py

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

./envs/large_grid_data