def check_classifier()

in workflow1_endpointbuilder/sam-app/functions/function7_checkclassifierstatus/index.py [0:0]

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

def check_classifier(classifier_arn):
    comprehend = boto3.client("comprehend")
    describe_response = comprehend.describe_document_classifier(
        DocumentClassifierArn=classifier_arn
    )

    is_trained = False
    status = describe_response['DocumentClassifierProperties']['Status']
    if status != "TRAINED":
        if status == "IN_ERROR":
            message = describe_response["DocumentClassifierProperties"]["Message"]
            raise ValueError(f"The classifier is in error:", message)
    else:
        is_trained = True

    return is_trained