combine_trees

jeanspy.sampler_numpyro.NumPyroSampler.combine_trees

Class and construction: NumPyroSampler.

static NumPyroSampler.combine_trees(trees)

Combine compatible ArviZ chunks.

Notes

Inputs and units. Nonempty sequence of DataTrees from the same analysis, with consistent groups, variables and nondraw coordinates.

Returns and shape. DataTree concatenated over draws, with draw coordinates renumbered from zero. No independent sampling-identity verification occurs here.

Validity. Supply chunks from the same analysis. Static groups must be equal. The draw-group concatenation uses xarray compat=override and is not a general identity validator.

Errors. ValueError for an empty list, missing groups, inconsistent draw-axis presence or changed static groups. Other xarray errors propagate.

Differentiation. Host-side sampling/storage control; this method has no physical-parameter derivative.

Parameters:

trees (Sequence[xarray.core.datatree.DataTree])

Return type:

xarray.core.datatree.DataTree