bartz.grove.check_trace¶

bartz.grove.check_trace(trace, max_split)[source]¶

Check the validity of a set of trees.

Use describe_error to parse the error codes returned by this function.

Parameters:

trace (TreeHeaps) – The set of trees to check. This object can have additional attributes beyond the tree arrays, they are ignored.
max_split (UInt[Array, 'p']) – The maximum split value for each variable.

Returns:

UInt[Array, '*batch_shape'] – A tensor of error codes for each tree.