flytekit.extend.ExecutionState

class flytekit.extend.ExecutionState(mode: Mode, working_dir: os.PathLike, engine_dir: os.PathLike, additional_context: Dict[Any, Any] = None)[source]
__init__(mode: Mode, working_dir: os.PathLike, engine_dir: os.PathLike, additional_context: Dict[Any, Any] = None)[source]

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(mode, working_dir, engine_dir[, …])

Initialize self.

branch_complete()

Indicates that we are within a conditional / ifelse block and the active branch is not done.

enter_conditional_section()

We cannot use a context manager here, so we will mimic the context manager API Reason we cannot use is because branch is a functional api and the context block is not well defined TODO we might want to create a new node manager here, as we want to capture all nodes in this branch context

exit_conditional_section()

Removes any current branch logic

take_branch()

Indicates that we are within an if-else block and the current branch has evaluated to true.

Attributes

additional_context

branch_eval_mode

engine_dir

mode

working_dir