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_adjusted_mutual_information"
                num: 0
              }
              custom_stats {
                name: "mean_adjusted_mutual_information"
                num: 0
              }
              custom_stats {
                name: "median_adjusted_mutual_information"
                num: 0
              }
              custom_stats {
                name: "min_adjusted_mutual_information"
                num: 0
              }
              custom_stats {
                name: "num_partitions_adjusted_mutual_information"
                num: 2.0
              }
              custom_stats {
                name: "std_dev_adjusted_mutual_information"
                num: 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 types
  from tensorflow_data_validation.statistics.generators import mutual_information
  from tensorflow_data_validation.statistics.generators import partitioned_stats_generator
  from tensorflow_data_validation.utils import test_util
  from tfx_bsl.arrow import table_util
  from google.protobuf import text_format
  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