def get_schema_entities(schema):
  """Gets the schema entities (column and table names) for a schema."""
  names = set()
  for table_name, cols in schema.items():
    names.add(table_name.lower().replace('_', ' '))
    for col in cols:
      names.add(col['field name'].lower().replace('_', ' '))
  return names
Gets the schema entities (column and table names) for a schema.
language/xsp/data_preprocessing/schema_utils.py