def parse_args()

in train/train.py [0:0]

• 5 lines of code
• 2 McCabe index (conditional complexity)

def parse_args():

    parser = argparse.ArgumentParser()
    # Data, model, and output directories
    # model_dir is always passed in from SageMaker. By default this is a S3 path under the default bucket
    parser.add_argument('--model-dir', type=str, default=os.environ.get('SM_MODEL_DIR'))
    parser.add_argument('--train', type=str, default=os.environ.get('SM_CHANNEL_TRAINING'))
    return parser.parse_known_args()