How to resolve Attribute Error in TensorFlow Extended

Question

import os
import pprint
import tempfile
import urllib

import absl
import tensorflow as tf
import tensorflow_model_analysis as tfma
tf.get_logger().propagate = False
pp = pprint.PrettyPrinter()

from tfx import v1 as tfx
from tfx.orchestration.experimental.interactive.interactive_context import InteractiveContext

%load_ext tfx.orchestration.experimental.interactive.notebook_extensions.skip

After executing the above code, I'm getting this error:

AttributeError: module 'tensorflow_model_analysis' has no attribute 'EvalConfig'

Can anyone explain why this is happening?