flytekit.clis.sdk_in_container.utils

Directory

Classes

Methods

Variables

Methods

get_option_from_metadata()

def get_option_from_metadata(
    metadata: mappingproxy,
) -> click.core.Option

make_click_option_field()

def make_click_option_field(
    o: click.core.Option,
) -> dataclasses.Field

pretty_print_exception()

def pretty_print_exception(
    e: Exception,
    verbosity: int = 1,
)

This method will print the exception in a nice way. It will also check if the exception is a grpc.RpcError and print it in a human-readable way.

pretty_print_grpc_error()

def pretty_print_grpc_error(
    e: grpc.RpcError,
)

This method will print the grpc error that us more human readable.

pretty_print_traceback()

def pretty_print_traceback(
    e: Exception,
    verbosity: int = 1,
)

This method will print the Traceback of an error. Print the traceback in a nice formatted way if verbose is set to True.

remove_unwanted_traceback_frames()

def remove_unwanted_traceback_frames(
    tb: traceback,
    unwanted_module_names: typing.List[str],
) -> traceback

Custom function to remove certain frames from the traceback.

validate_package()

def validate_package(
    ctx,
    param,
    values,
)

This method will validate the packages passed in by the user. It will check that the packages are in the correct format, and will also split the packages if the user passed in a comma separated list.

flytekit.clis.sdk_in_container.utils.ErrorHandlingCommand

Helper class that wraps the invoke method of a click command to catch exceptions and print them in a nice way.

Parameters

class ErrorHandlingCommand(
    *args: Any,
    **kwargs: Any,
)

Create RichGroup instance.

Methods

invoke()

def invoke(
    ctx: click.core.Context,
) -> typing.Any

Given a context, this invokes the attached callback (if it exists) in the right way.

flytekit.clis.sdk_in_container.utils.PyFlyteParams

Parameters

class PyFlyteParams(
    config_file: typing.Optional[str] = None,
    verbose: bool = False,
    pkgs: typing.List[str] = <factory>,
)

Methods

from_dict()

def from_dict(
    d: typing.Dict[str, typing.Any],
) -> PyFlyteParams