code: def get_dataset_provider():
      """Helper function to get the data provider."""
      logging.info("Getting the registered data provider")
      # Reigstration API
      data_providers = registry.lookup_all(ms_data.Provider)
      if len(data_providers) == 1:
          return data_providers[0]
  
      # Registering more than one data provider
      else:
          logging.info(
              "Registered data provider: %s", FLAGS.phoenix_dataset
          )
          return registry.lookup(
              FLAGS.phoenix_dataset, ms_data.Provider
          )
  
imports: import contextlib
  import copy
  import json
  import os
  import random
  from absl import flags
  from absl import logging
  import keras_tuner
  from model_search import hparam
  from model_search import phoenix
  from model_search import registry
  from model_search.data import data
  from model_search.metadata import ml_metadata_db
  from model_search.proto import phoenix_spec_pb2
  import tensorflow.compat.v2
  from google.protobuf import text_format
label_desc: Use image node_id 2 for calling a global function with example usage: get_dataset_provider() and returns: data_providers, registry