SageMaker connector

The SageMaker connector allows you to deploy models, and create and trigger inference endpoints. You can also fully remove the SageMaker deployment.

Installation

To use the SageMaker connector, run the following command:

$ pip install flytekitplugins-awssagemaker

Example usage

Local testing

To test an connector locally, create a class for the connector task that inherits from SyncConnectorExecutorMixin or AsyncConnectorExecutorMixin. These mixins can handle synchronous and synchronous tasks, respectively, and allow the SDK to mimic the system’s behavior in calling the connector.

For more information, see Testing connectors locally.