def main()

in benchmark/tensorflow/generate_runner.py [0:0]

• 28 lines of code
• 17 McCabe index (conditional complexity)

def main(argv):

    try:
        opts, args = getopt.getopt(argv, "", ["model=","nodes=","remote_dir=","gen_dir=","num_nodes=","gpu_per_node=","batch_size="])
    except getopt.GetoptError:
        print("Incorrect args")
        sys.exit(2)
    
    for opt, arg in opts:
        if opt == "--model":
            model = arg
        if opt == "--nodes":
            nodes_file = os.path.abspath(arg)
        if opt == "--remote_dir":
            remote_dir = arg
        elif opt == "--gen_dir":
            gen_dir = os.path.abspath(arg)
            gen_dir_rel = arg
        if opt == "--num_nodes":
            num_nodes = int(arg)
        elif opt == "--gpu_per_node":
            gpu_per_node = int(arg)
        elif opt == "--batch_size":
            batch_size = int(arg)
    
    if(remote_dir == None or gen_dir == None or num_nodes == None or gpu_per_node == None or batch_size == None or nodes_file == None or model == None):
        print("Incorrect args")
        sys.exit(2)
    
    script_name = get_script_name(model)
        
    gen_scripts(model, nodes_file, remote_dir, gen_dir_rel, num_nodes, gpu_per_node, batch_size)
    
    print(script_name)