def parse_args()

in src/responsibleai/rai_analyse/create_rai_insights.py [0:0]

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

def parse_args():
    # setup arg parser
    parser = argparse.ArgumentParser()

    parser.add_argument("--title", type=str, required=True)

    parser.add_argument(
        "--task_type", type=str, required=True, choices=["classification", "regression", "forecasting"]
    )

    parser.add_argument("--train_dataset", type=str, required=True)
    parser.add_argument("--test_dataset", type=str, required=True)

    parser.add_argument("--target_column_name", type=str, required=True)

    parser.add_argument("--maximum_rows_for_test_dataset", type=int, default=5000)

    parser.add_argument(
        "--categorical_column_names", type=str, help="Optional[List[str]]"
    )

    parser.add_argument("--model_info_path", type=str, help="name:version")

    parser.add_argument("--model_input", type=str, help="model local path on remote")

    parser.add_argument("--model_info", type=str, help="name:version")

    parser.add_argument("--classes", type=str, help="Optional[List[str]]")

    parser.add_argument(
        "--feature_metadata",
        type=str,
        help="identity_feature_name:Optional[str], "
             "dropped_features:Optional[List[str]], "
             "datetime_features:Optional[List[str]], "
             "time_series_id_features:Optional[List[str]]",
    )

    parser.add_argument(
        "--use_model_dependency", type=boolean_parser, help="Use model dependency"
    )

    parser.add_argument("--output_path", type=str, help="Path to output JSON")

    # parse args
    args = parser.parse_args()

    # return args
    return args