free_effect_types

penzai.data_effects.effect_base.free_effect_types(model_tree: Any) → list[type[Any]]

Collects the effect types of all EffectRequest nodes in a (sub)model.

Parameters:

model_tree – A pytree for a model or submodel.

Returns:

A list of the effect protocol types for all EffectRequest nodes that appear in a model or submodel.