def parse_args()

in sagemaker_notebook_instance/containers/model/entry_point.py [0:0]

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

def parse_args(sys_args):
    parser = argparse.ArgumentParser()

    parser.add_argument(
        "--n-estimators",
        type=int,
        default=100
    )
    parser.add_argument(
        "--model-dir",
        type=str,
        default=os.environ.get("SM_MODEL_DIR")
    )
    parser.add_argument(
        "--data-train",
        type=str,
        default=os.environ.get("SM_CHANNEL_TRAIN"),
    )
    parser.add_argument(
        "--data-test",
        type=str,
        default=os.environ.get("SM_CHANNEL_TEST")
    )
    parser.add_argument(
        "--numerical-feature-names",
        type=str
    )
    parser.add_argument(
        "--categorical-feature-names",
        type=str
    )
    parser.add_argument(
        "--textual-feature-names",
        type=str
    )
    parser.add_argument(
        "--label-name",
        type=str
    )

    args, _ = parser.parse_known_args(sys_args)
    print(args)
    return args