traveler.core.array_tools.JaxTable.assign_in_place

JaxTable.assign_in_place(vars=None, **kw_vars)

Assign new variables to the JaxDataset, in place.

This method allows you to add new fields to the dataset. If the field already exists, it will be overwritten. Usually, we should prefer assign to return a new instance.

Parameters:

vars (dict[str, Any] | None)

Return type:

None