def main()

in comprehend_groundtruth_integration/src/comprehend_customer_scripts/GroundTruth/DocumentClassifier/groundtruth_format_conversion_handler.py [0:0]

• 14 lines of code
• 3 McCabe index (conditional complexity)

def main():
    parser = argparse.ArgumentParser(description="Parsing the output S3Uri")
    parser.add_argument('mode')
    parser.add_argument('dataset_output_S3Uri')
    parser.add_argument('label_delimiter')
    args = parser.parse_args()
    handler = GroundTruthToCLRFormatConversionHandler()
    handler.validate_s3_input(args)
    if args.mode == "MULTI_CLASS":
        handler.read_write_multiclass_dataset()
    elif args.mode == "MULTI_LABEL":
        handler.read_write_multilabel_dataset(args.label_delimiter)
    else:
        raise Exception("The value provided for mode is invalid. Valid values are MUTLI_CLASS|MULTI_LABEL")