tbp.monty.attention#

tbp.monty.attention.attention_system#

class AttentionSystemProtocol(*args, **kwargs)[source]#

Bases: Protocol

__init__(*args, **kwargs)#

reset() → None[source]#

Return type:: None

state_dict() → Mapping[str, Any][source]#

Return type:: Mapping[str, Any]

step(goals: Sequence[Goal], regions: Sequence[AttentionRegion]) → list[Goal][source]#

class NoopAttentionSystem(*args, **kwargs)[source]#

Bases: AttentionSystemProtocol

reset() → None[source]#

Nothing to reset.

Return type:: None

state_dict() → Mapping[str, Any][source]#

Return type:: Mapping[str, Any]

step(goals: Sequence[Goal], regions: Sequence[AttentionRegion]) → list[Goal][source]#