ModuleNotFoundError: No module named 'tfx.utils.dsl_utils'

I did install tfx version 1.2.1Python: 3.8.2Tensorflow: 2.5.2pip: 21.3.1I use window and installed the package through pip.

The error occurred when I did:

import osfrom tfx.components import CsvExampleGenfrom tfx.utils.dsl_utils import external_inputbase_dir = os.getcwd()data_dir = os.path.join(os.pardir, "data")examples = external_input(os.path.join(base_dir, data_dir))example_gen = CsvExampleGen(input=examples)context.run(example_gen)

The error:

ModuleNotFoundError                       Traceback (most recent call last)~\AppData\Local\Temp/ipykernel_14604/719215164.py in <module>      1 import os      2 from tfx.components import CsvExampleGen----> 3 from tfx.utils.dsl_utils import external_input      4 base_dir = os.getcwd()      5 data_dir = os.path.join(os.pardir, "data")ModuleNotFoundError: No module named 'tfx.utils.dsl_utils'

I did full installation of tfx in which all packages are compatible. Any help is appreciated.