flytekit.utils.dict_formatter

Directory

Methods

Methods

format_dict()

def format_dict(
    service: str,
    original_dict: typing.Any,
    update_dict: typing.Dict[str, typing.Any],
    idempotence_token: typing.Optional[str],
) -> typing.Any

Recursively update a dictionary with format strings with values from another dictionary where the keys match the format string. This goes a little beyond regular python string formatting and uses . to denote nested keys.

For example, if original_dict is {“EndpointConfigName”: “{endpoint_config_name}”}, and update_dict is {“endpoint_config_name”: “my-endpoint-config”}, then the result will be {“EndpointConfigName”: “my-endpoint-config”}.

For nested keys if the original_dict is {“EndpointConfigName”: “{inputs.endpoint_config_name}”}, and update_dict is {“inputs”: {“endpoint_config_name”: “my-endpoint-config”}}, then the result will be {“EndpointConfigName”: “my-endpoint-config”}.

Returns: The updated dictionary

get_nested_value()

def get_nested_value(
    d: typing.Dict[str, typing.Any],
    keys: list[str],
) -> typing.Any

Retrieve the nested value from a dictionary based on a list of keys.

replace_placeholder()

def replace_placeholder(
    service: str,
    original_dict: str,
    placeholder: str,
    replacement: str,
) -> str

Replace a placeholder in the original string and handle the specific logic for the sagemaker service and idempotence token.