opto.trace.propagators.graph_propagator#

Module Contents#

Classes#

TraceGraph

Feedback container used by GraphPropagator.

GraphPropagator

A propagator that collects all the nodes seen in the path.

API#

class TraceGraph[source]#

Bases: opto.trace.propagators.propagators.AbstractFeedback

Feedback container used by GraphPropagator.

graph: List[Tuple[int, opto.trace.nodes.Node]] = None#
user_feedback: Any = None#
classmethod expand(node: opto.trace.nodes.MessageNode)[source]#

Return the subgraph within a MessageNode.

visualize(simple_visualization=True, reverse_plot=False, print_limit=100)[source]#
class GraphPropagator[source]#

Bases: opto.trace.propagators.propagators.Propagator

A propagator that collects all the nodes seen in the path.

Initialization

init_feedback(node, feedback: Any)[source]#
aggregate(feedback: Dict[opto.trace.nodes.Node, List[opto.trace.propagators.graph_propagator.TraceGraph]])[source]#

Aggregate feedback from multiple children