flytekitplugins.awssagemaker_inference.create_sagemaker_deployment

flytekitplugins.awssagemaker_inference.create_sagemaker_deployment(name, model_config, endpoint_config_config, endpoint_config, images=None, model_input_types=None, endpoint_config_input_types=None, endpoint_input_types=None, region=None, region_at_runtime=False)[source]

Creates SageMaker model, endpoint config and endpoint.

Parameters:
  • model_config (Dict[str, Any]) – Configuration for the SageMaker model creation API call.

  • endpoint_config_config (Dict[str, Any]) – Configuration for the SageMaker endpoint configuration creation API call.

  • endpoint_config (Dict[str, Any]) – Configuration for the SageMaker endpoint creation API call.

  • images (Dict[str, Any] | None) – A dictionary of images for SageMaker model creation.

  • model_input_types (Dict[str, Type] | None) – Mapping of SageMaker model configuration inputs to their types.

  • endpoint_config_input_types (Dict[str, Type] | None) – Mapping of SageMaker endpoint configuration inputs to their types.

  • endpoint_input_types (Dict[str, Type] | None) – Mapping of SageMaker endpoint inputs to their types.

  • region (str | None) – The region for SageMaker API calls.

  • region_at_runtime (bool) – Set this to True if you want to provide the region at runtime.

  • name (str)

Return type:

ImperativeWorkflow