def infer()

in cvat-serverless/functions/endpoints/pth_rcnn.py [0:0]

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

def infer(image):

    client = boto3.client('runtime.sagemaker')

    response = client.invoke_endpoint(EndpointName=settings.PTH_RCNN_ENDPOINT,
                                      Body=image,
                                      TargetContainerHostname=settings.PTH_RCNN_NAME,
                                      ContentType='application/x-image')
    response_body = response['Body'].read()
    result = json.loads(response_body.decode('utf-8'))

    return result