def call_custom_comprehend_model()

in files/lambda_code/call_cm_other_models_lambda.py [0:0]

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

def call_custom_comprehend_model(the_input=None,endpoint_arn=None):
    '''call the custom comprehend model that has been previously trained to classify the document according to its medical specialty type.'''
    client = boto3.client('comprehend')
    response = client.classify_document(
        Text=the_input,
        EndpointArn=endpoint_arn
        )
    return(response)