traveler.templates.persons.Persons.to_dict

Persons.to_dict()

Convert the JaxDataset to a dictionary of JAX arrays.

CategoricalArrays are converted to pandas Series instead of JAX arrays.

Return type:

dict[str, jt.Array | pd.Series]