vertex_pipelines_simple lab breaks at imports

[jasonbrancazio]

https://github.com/GoogleCloudPlatform/training-data-analyst/blob/master/self-paced-labs/tfx/tfx-vertex/vertex_pipelines_simple.ipynb

Running via https://partner.cloudskillsboost.google/course_templates/666

pip dependency resolution seems to have gone off the rails, leading to TF 1.15 being installed, and then the following error:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
/tmp/ipykernel_12292/1194142330.py in <module>
      1 import tensorflow as tf
      2 print('TensorFlow version: {}'.format(tf.__version__))
----> 3 from tfx import v1 as tfx
      4 print('TFX version: {}'.format(tfx.__version__))
      5 import kfp

/opt/conda/lib/python3.7/site-packages/tfx/v1/__init__.py in <module>
     18 try:
     19   # These modules may not be available in some environments.
---> 20   from tfx.v1 import components
     21   from tfx.v1 import extensions
     22   from tfx.v1 import orchestration

/opt/conda/lib/python3.7/site-packages/tfx/v1/components/__init__.py in <module>
     27 from tfx.components.statistics_gen.component import StatisticsGen
     28 from tfx.components.trainer.component import Trainer
---> 29 from tfx.components.transform.component import Transform
     30 from tfx.components.tuner.component import Tuner
     31 

/opt/conda/lib/python3.7/site-packages/tfx/components/transform/component.py in <module>
     17 
     18 from tfx import types
---> 19 from tfx.components.transform import executor
     20 from tfx.components.util import udf_utils
     21 from tfx.dsl.components.base import base_beam_component

/opt/conda/lib/python3.7/site-packages/tfx/components/transform/executor.py in <module>
     23 import tensorflow as tf
     24 import tensorflow_data_validation as tfdv
---> 25 import tensorflow_transform as tft
     26 from tensorflow_transform import impl_helper
     27 import tensorflow_transform.beam as tft_beam

/opt/conda/lib/python3.7/site-packages/tensorflow_transform/__init__.py in <module>
     34 # filesystem.
     35 try:
---> 36   import tensorflow_io as _  # pytype: disable=import-error # pylint: disable=g-import-not-at-top
     37 except ModuleNotFoundError:
     38   pass

/opt/conda/lib/python3.7/site-packages/tensorflow_io/__init__.py in <module>
     15 """tensorflow_io"""
     16 
---> 17 from tensorflow_io.python.api import *  # pylint: disable=wildcard-import
     18 from tensorflow_io.python.api.version import VERSION as __version__

/opt/conda/lib/python3.7/site-packages/tensorflow_io/python/api/__init__.py in <module>
     17 
     18 # tensorflow_io.core.python.ops is implicitly imported (along with file system)
---> 19 from tensorflow_io.python.ops.io_dataset import IODataset
     20 from tensorflow_io.python.ops.io_tensor import IOTensor
     21 

/opt/conda/lib/python3.7/site-packages/tensorflow_io/python/ops/__init__.py in <module>
     94 core_ops = LazyLoader("core_ops", "libtensorflow_io.so")
     95 try:
---> 96     plugin_ops = _load_library("libtensorflow_io_plugins.so", "fs")
     97 except NotImplementedError as e:
     98     warnings.warn("unable to load libtensorflow_io_plugins.so: {}".format(e))

/opt/conda/lib/python3.7/site-packages/tensorflow_io/python/ops/__init__.py in _load_library(filename, lib)
     62     for f in filenames:
     63         try:
---> 64             l = load_fn(f)
     65             if l is not None:
     66                 return l

/opt/conda/lib/python3.7/site-packages/tensorflow_io/python/ops/__init__.py in <lambda>(f)
     54         load_fn = lambda f: ctypes.CDLL(f, mode=ctypes.RTLD_GLOBAL)
     55     elif lib == "fs":
---> 56         load_fn = lambda f: tf.experimental.register_filesystem_plugin(f) is None
     57     else:
     58         load_fn = lambda f: tf.compat.v1.load_file_system_library(f) is None

/opt/conda/lib/python3.7/site-packages/tensorflow_core/python/util/module_wrapper.py in __getattr__(self, name)
    191   def __getattr__(self, name):
    192     try:
--> 193       attr = getattr(self._tfmw_wrapped_module, name)
    194     except AttributeError:
    195       if not self._tfmw_public_apis:

AttributeError: module 'tensorflow._api.v1.experimental' has no attribute 'register_filesystem_plugin'