def parse_args()

in deep_learning_container.py [0:0]

• 16 lines of code
• 1 McCabe index (conditional complexity)

def parse_args():
    """
    Parsing function to parse input arguments.
    Return: args, which containers parsed input arguments.
    """
    parser = argparse.ArgumentParser()
    parser.add_argument("--framework",
                        choices=["tensorflow", "mxnet", "pytorch"],
                        help="framework of container image.",
                        required=True)
    parser.add_argument("--framework-version",
                        help="framework version of container image.",
                        required=True)
    parser.add_argument("--container-type",
                        choices=["training", "inference"],
                        help="What kind of jobs you want to run on container. \
                        Either training or inference.",
                        required=True)

    args, _unknown = parser.parse_known_args()

    return args