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.MinimalTrace 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. Implementations must be equinox.Module dataclasses whose per-sample fields carry samples axis markers (see bartz.mcmcstep._axes.field), read by evaluate_trace to locate the sample axis.