code: def _get_test_stats_with_mi(feature_paths):
      """Get stats proto for MI test."""
      result = statistics_pb2.DatasetFeatureStatistics()
      for feature_path in feature_paths:
          feature_proto = text_format.Parse(
              """
                  custom_stats {
                    name: "max_sklearn_adjusted_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "max_sklearn_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "max_sklearn_normalized_adjusted_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "mean_sklearn_adjusted_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "mean_sklearn_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "mean_sklearn_normalized_adjusted_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "median_sklearn_adjusted_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "median_sklearn_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "median_sklearn_normalized_adjusted_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "min_sklearn_adjusted_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "min_sklearn_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "min_sklearn_normalized_adjusted_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "num_partitions_sklearn_adjusted_mutual_information"
                    num: 2.0
                  }
                  custom_stats {
                    name: "num_partitions_sklearn_mutual_information"
                    num: 2.0
                  }
                  custom_stats {
                    name: "num_partitions_sklearn_normalized_adjusted_mutual_information"
                    num: 2.0
                  }
                  custom_stats {
                    name: "std_dev_sklearn_adjusted_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "std_dev_sklearn_mutual_information"
                    num: 0.0
                  }
                  custom_stats {
                    name: "std_dev_sklearn_normalized_adjusted_mutual_information"
                    num: 0.0
                  }
          """,
              statistics_pb2.FeatureNameStatistics(),
          )
          feature_proto.path.CopyFrom(feature_path.to_proto())
          result.features.add().CopyFrom(feature_proto)
      return result
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import absltest
  from absl.testing import parameterized
  import apache_beam
  from apache_beam.testing import util
  import numpy
  import pyarrow
  from tensorflow_data_validation import constants
  from tensorflow_data_validation import types
  from tensorflow_data_validation.statistics.generators import partitioned_stats_generator
  from tensorflow_data_validation.statistics.generators import sklearn_mutual_information
  from tensorflow_data_validation.utils import test_util
  from google.protobuf import text_format
  from tensorflow.python.util.protobuf import compare
  from tensorflow_metadata.proto.v0 import schema_pb2
  from tensorflow_metadata.proto.v0 import statistics_pb2
label_desc: Use image node_id 1 for calling a global function with example usage: _get_test_stats_with_mi(feature_paths) and returns: result