step_in_line.step

module documentation

(source)

Undocumented

Class	`Step`	`Step` for step function workflows
Function	`step`	Decorator for converting a python function to a pipeline step.

def step(_func=None, *, name: Optional[str] = None, description: Optional[str] = None, layers: Optional[List[str]] = None, python_runtime: str = 'python3.10', memory_size: int = 512, policies: List[str] = [], retry_count: int = 0, env_variables: Dict[str, str] = {}): (source) ¶

Decorator for converting a python function to a pipeline step.

This decorator wraps the annotated code into a Step object which can then be passed to a pipeline as a step.

Args:: _func: A Python function to run as a SageMaker pipeline step. name (str): Name of the pipeline step. Defaults to a generated name using function name and uuid4 identifier to avoid duplicates. description (str): Description of the step layers (list): Lambda layers python_runtime (str): Lambda runtime. memory_size (int): Megabytes of memory for the lambda. Defaults to 512. policies (List[str]): IAM policies, in JSON, to provide to the Lambda retry_count (int): number of retries to attempt. Defaults to 0 (no retries). env_variables (dict): environment variables to pass to lambda function