def get_atoms_with_num_arguments(target):
  """Consider symbols and their number of arguments."""
  name, arguments = _get_name_and_arguments(target)
  if arguments:
    atoms = set()
    atoms.add("%s_(%s)" % (name, len(arguments)))
    for argument in arguments:
      atoms |= get_atoms_with_num_arguments(argument)
    return atoms
  else:
    return {name}
Consider symbols and their number of arguments.
language/compgen/nqg/tasks/geoquery/tmcd_utils.py