def _find_simple_entity(token,
                        example,
                        include_tables=True,
                        include_columns=True):
  """Finds entities in the schema that a token may be referring to."""
  matching = _find_all_entities(token, example, include_tables, include_columns)

  if not matching:
    return None

  if len(matching) == 1:
    return matching[0]
Finds entities in the schema that a token may be referring to.
language/xsp/data_preprocessing/sql_parsing.py