sagemaker.mlops.workflow.callback_step

The step definitions for workflow.

Classes

CallbackOutput([output_name, output_type])	Output for a callback step.
CallbackOutputTypeEnum(*args[, value])	CallbackOutput type enum.
CallbackStep(name, sqs_queue_url, inputs, ...)	Callback step for workflow.

class sagemaker.mlops.workflow.callback_step.CallbackOutput(output_name: str | None = None, output_type: CallbackOutputTypeEnum = CallbackOutputTypeEnum.String)

Bases: object

Output for a callback step.

output_name

The output name

Type:

str

output_type

The output type

Type:

CallbackOutputTypeEnum

expr(step_name) → Dict[str, str]

The ‘Get’ expression dict for a CallbackOutput.

output_name: str

output_type: CallbackOutputTypeEnum

to_request() → Dict[str, Any] | List[Dict[str, Any]]

Get the request structure for workflow service calls.

class sagemaker.mlops.workflow.callback_step.CallbackOutputTypeEnum(*args, value=<object object>, **kwargs)

Bases: Enum

CallbackOutput type enum.

Boolean = 'Boolean'

Float = 'Float'

Integer = 'Integer'

String = 'String'

class sagemaker.mlops.workflow.callback_step.CallbackStep(name: str, sqs_queue_url: str, inputs: dict, outputs: List[CallbackOutput], display_name: str | None = None, description: str | None = None, cache_config: CacheConfig | None = None, depends_on: List[str | Step] | None = None)

Bases: Step

Callback step for workflow.

property arguments: Dict[str, Any] | List[Dict[str, Any]]

The arguments dict that is used to define the callback step.

property properties

A Properties object representing the output parameters of the callback step.

to_request() → Dict[str, Any] | List[Dict[str, Any]]

Updates the dictionary with cache configuration.