code: def get_categorical_features(
      schema: schema_pb2.Schema,
  ) -> Set[types.FeaturePath]:
      """Gets the set containing the names of all categorical features.
  
      Args:
        schema: The schema for the data.
  
      Returns:
        A set containing the names of all categorical features.
      """
      return {
          feature_path
          for feature_path, feature in get_all_leaf_features(schema)
          if is_categorical_feature(feature)
      }
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  import logging
  from typing import Any, Iterable, List, Mapping, Optional, Set, Text, Tuple, Union
  from tensorflow_data_validation import types
  from tensorflow_data_validation.utils import io_util
  from google.protobuf import descriptor
  from google.protobuf import text_format
  from tensorflow_metadata.proto.v0 import schema_pb2
label_desc: Use image node_id 10 for calling a global function with example usage: get_categorical_features(schema) and returns: unknown