flytekitplugins.awssagemaker_inference.SageMakerEndpointAgent

class flytekitplugins.awssagemaker_inference.SageMakerEndpointAgent

This agent creates an endpoint.

Methods

async create(task_template, inputs=None, **kwargs)

Return a resource meta that can be used to get the status of the task.

Parameters:

• task_template (TaskTemplate)

• inputs (LiteralMap | None)

Return type:

SageMakerEndpointMetadata

async delete(resource_meta, **kwargs)

Delete the task. This call should be idempotent. It should raise an error if fails to delete the task.

Parameters:

resource_meta (SageMakerEndpointMetadata)

async get(resource_meta, **kwargs)

Return the status of the task, and return the outputs in some cases. For example, bigquery job can’t write the structured dataset to the output location, so it returns the output literals to the propeller, and the propeller will write the structured dataset to the blob store.

Parameters:

resource_meta (SageMakerEndpointMetadata)

Return type:

Resource

Attributes

metadata_type

name = 'SageMaker Endpoint Agent'

task_category

task category that the agent supports