docs: 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