code: def extract_keys(feature_dict, feature_cls):
      """Extracts keys from features dict based on feature type.
  
      Args:
        feature_dict: `tfds.features.FeaturesDict` from which extract keys
        feature_cls: `tfds.features.FeatureConnector` class to search.
  
      Returns:
        List of extracted keys matching the class.
      """
      return [
          k
          for k, f in feature_dict.items()
          if isinstance(f, feature_cls)
      ]
  
file_path: datasets/tensorflow_datasets/core/visualization/visualizer.py
imports: from __future__ import annotations
  import abc
  from typing import Any
  from tensorflow_datasets.core import dataset_info
  from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: extract_keys(feature_dict, feature_cls) and returns: unknown