pymc.backends.zarr.ZarrChain.record_sampling_state

ZarrChain.record_sampling_state(step=None)

Record the sampling state information to the store’s _sampling_state group.

The sampling state includes the number of draws taken so far (draw_idx) and the step method’s sampling_state.

Parameters:

stepBlockedStep | CompoundStep | None

The step method from which to take the sampling_state. If None, the step is taken to be the step method that was linked to the ZarrChain when calling link_stepper(). If this method was never called, no step method sampling_state information is stored in the chain.