def process_node_args()

in vihds/inference_graph.py [0:0]

• 31 lines of code
• 12 McCabe index (conditional complexity)

def process_node_args(name, yamlargs, graph_name):
    split = False
    argarr = []
    # Should this also see if any one of those 3 arguments are in the command line?
    if "split" in yamlargs:
        True
        argarr.append("--split=" + str(yamlargs.split))

    if "spec" in yamlargs:
        argarr.append(yamlargs.spec)
    else:
        raise ValueError("Node " + name + " missing spec property")

    if "experiment" in yamlargs:
        argarr.append("--experiment=" + graph_name + "/" + yamlargs.experiment)
    else:
        raise ValueError("Node " + name + " missing experiment property")

    if "seed" in yamlargs:
        argarr.append("--seed=" + str(yamlargs.seed))

    if "train_samples" in yamlargs:
        argarr.append("--train_samples=" + str(yamlargs.train_samples))

    if "test_samples" in yamlargs:
        argarr.append("--test_samples=" + str(yamlargs.test_samples))

    if "epochs" in yamlargs:
        argarr.append("--epochs=" + str(yamlargs.epochs))

    if "test_epoch" in yamlargs:
        argarr.append("--test_epoch=" + str(yamlargs.test_epoch))

    if "plot_epoch" in yamlargs:
        argarr.append("--plot_epoch=" + str(yamlargs.plot_epoch))

    if "gpu" in yamlargs:
        argarr.append("--gpu=" + str(yamlargs.gpu))

    # Should probably add other arguments as well...

    parser = rxval.create_parser(split)
    args = parser.parse_args(argarr)

    return args