code: def _get_dense_features_module():
      """Returns the module that contains DenseFeatures class."""
      try:
          return tf.compat.v1.keras.layers
      except:  # pylint: disable=bare-except
          return feature_column_v2
  
imports: import contextlib
  import inspect
  import os
  import pickle
  from typing import Any, Callable, Dict, Optional
  import six
  import tensorflow
  from tensorflow import estimator
  import tensorflow_model_analysis
  from tensorflow_model_analysis.api import types
  from tensorflow_model_analysis.utils import util
  from tensorflow.python.feature_column import feature_column_v2
  from tensorflow_estimator.python.estimator.canned import dnn
label_desc: Use image node_id 3 for calling a global function with example usage: _get_dense_features_module() and returns: tf, feature_column_v2