lifecycle.api.EdgeConnectionHook

class hamilton.lifecycle.api.EdgeConnectionHook[source]

Implement this to customize edges that are allowed in the graph. You can do customizations around typing here.

abstractmethod check_edge_types_match(type_from: type, type_to: type, **kwargs: Any) → bool[source]

This is run to check if edge types match. Note that this is an OR functionality – this is run after we do some default checks, so this can only be permissive. Reach out if you want to be more restrictive than the default checks.

Parameters:

• type_from – The type of the node that is the source of the edge.
• type_to – The type of the node that is the destination of the edge.
• kwargs – This is kept for future backwards compatibility.

Returns:

Whether or not the two node types form a valid edge.

final do_check_edge_types_match(*, type_from: type, type_to: type) → bool[source]

Wraps the check_edge_types_match method, providing a bridge to an external-facing API. Do not override this!

final do_validate_input(*, node_type: type, input_value: Any) → bool[source]

Wraps the validate_input method, providing a bridge to an external-facing API. Do not override this!

abstractmethod validate_input(node_type: type, input_value: Any, **kwargs: Any) → bool[source]

This is run to check if the input is valid for the node type. Note that this is an OR functionality – this is run after we do some default checks, so this can only be permissive. Reach out if you want to be more restrictive than the default checks.

Parameters:

• node_type – Type of the node that is accepting the input.
• input_value – Value of the input
• kwargs – Keyword arguments – this is kept for future backwards compatibility.

Returns:

Whether the input is valid for the node type.