pymc.backends.zarr.ZarrTrace.split_warmup — PyMC 5.22.0 documentation (original) (raw)

ZarrTrace.split_warmup(group_name, error_if_already_split=True)[source]#

Split the arrays of a group into the warmup and regular groups.

This function takes the first self.tuning_steps draws of suppliedgroup_name and moves them into a new zarr group calledf"warmup_{group_name}".

Parameters:

group_namestr

The name of the group that should be split.

error_if_already_splitbool

If True and if the f"warmup_{group_name}" group already exists in the root hierarchy, a ValueError is raised. If this flag is Falsebut the warmup group already exists, the contents of that group are overwritten.