def METHOD_NAME(
    self,
    grid: Grid,
    flow_field: FlowField,
) -> Dict[str, Any]:
    kwargs = {
        "x": grid.x_sorted,
        "y": grid.y_sorted,
        "z": grid.z_sorted,
        "u_initial": flow_field.u_initial_sorted,
    }
    return kwargs