Split data into multiple timesteps.

def split_timesteps(data, consistent_abmn=False):
    """Split data into multiple timesteps."""
    if has_multiple_timesteps(data):
        grouped = data.groupby("timestep")
        return [group[1] for group in grouped]
    else:
        return data