import boto3 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 def lambda_handler(event, context): datetime_id = event["datetime_id"] classifier_arn = event["classifier_arn"] is_trained = check_classifier(classifier_arn) return { 'is_trained': is_trained, 'datetime_id': datetime_id, 'classifier_arn': classifier_arn }