traveler.templates.LandUseTAZ.to_dict#

LandUseTAZ.to_dict()[source]#

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]