orbit.actions.ConditionalAction

Represents an action that is only taken when a given condition is met.

orbit.actions.ConditionalAction(
    condition: Condition,
    action: Union[controller.Action, Sequence[controller.Action]]
)

This class is itself an Action (a callable that can be applied to train or eval outputs), but is intended to make it easier to write modular and reusable conditions by decoupling "when" something whappens (the condition) from "what" happens (the action).

condition A callable accepting train or eval outputs and returning a bool.
action The action (or optionally sequence of actions) to perform when condition is met.

Methods

__call__

View source

__call__(
    output: runner.Output
) -> None

Call self as a function.