def update_endpoint_arn()

in backend/training-pipeline/functions/step_functions/delete_oldest_endpoint.py [0:0]

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

def update_endpoint_arn(arn, status):
    response = model_table.query(
        IndexName='endpointArn',
        KeyConditionExpression=Key('endpointArn').eq(arn)
    )
    items = response['Items']
    if len(items) > 0:
        item = items[0]
        model_table.update_item(
            Key={
                'trainingId': item['trainingId'],
                'created': item['created']
            },
            UpdateExpression='SET #st = :ns',
            ExpressionAttributeNames={
                '#st': 'status'
            },
            ExpressionAttributeValues={
                ':ns': status
            }
        )