traveler.core.stored.ReferencedTable.with_index_array#

ReferencedTable.with_index_array(index_array)[source]#

Set the index array for a copy of this ReferencedTable.

Parameters:

index_array (jax.numpy.ndarray | CategoricalArray) – The index array to use for this ReferencedTable.

Returns:

A new instance of the ReferencedTable with the given index array.

Return type:

ReferencedTable