bartz.mcmcloop.EvaluableTrace

class bartz.mcmcloop.EvaluableTrace(*args, **kwargs)[source]

Structural type of the traces accepted by evaluate_trace.

Both bartz.mcmcloop.MainTrace and bartz.debug.TraceWithOffset satisfy it. The runtime check is structural (attribute presence only, not the annotated shapes), so it also matches the axis-spec trees chain_vmap_axes derives from a trace.