def nth_child_edge_types(max_child_count):
  """Constructs the edge types for nth-child edges.

  Args:
    max_child_count: Maximum number of children that get explicit nth-child
      edges.

  Returns:
    Set of edge type names.
  """
  return {f"CHILD_INDEX_{i}" for i in range(max_child_count)}