Import errors when trying to run Chicago Taxi on Dataflow about tfx HOT 14 CLOSED

import setuptools

# LINT.IfChange
TF_VERSION = '1.12.0'
# LINT.ThenChange(train_mlengine.sh, start_model_server_mlengine.sh)

# LINT.IfChange
BEAM_VERSION = '2.11.0'
# LINT.ThenChange(setup_beam_on_flink.sh)

if __name__ == '__main__':
  setuptools.setup(
      name='tfx_chicago_taxi',
      version='0.12.0',
      packages=setuptools.find_packages(),
      install_requires=[
          'apache-beam[gcp]==' + BEAM_VERSION,
          'jupyter==1.0',
          'numpy==1.14.5',
          'protobuf==3.6.1',
          'tensorflow==' + TF_VERSION,
          'tensorflow-data-validation==0.12.0',
          'tensorflow-metadata==0.12.1',
          'tensorflow-model-analysis==0.12.1',
          'tensorflow-serving-api==1.12.0',
          'tensorflow-transform==0.12.0',
      ],
      python_requires='>=2.7,<3')

Traceback (most recent call last):
  File "tfdv_analyze_and_validate.py", line 191, in <module>
    main()
  File "tfdv_analyze_and_validate.py", line 176, in main
    pipeline_args=pipeline_args)
  File "tfdv_analyze_and_validate.py", line 121, in compute_stats
    statistics_pb2.DatasetFeatureStatisticsList)))
  File "/Users/gogasca/Documents/Development/dpe/tfx_taxi_cloud/taxi/lib/python2.7/site-packages/apache_beam/pipeline.py", line 426, in __exit__
    self.run().wait_until_finish()
  File "/Users/gogasca/Documents/Development/dpe/tfx_taxi_cloud/taxi/lib/python2.7/site-packages/apache_beam/runners/dataflow/dataflow_runner.py", line 1238, in wait_until_finish
    (self.state, getattr(self._runner, 'last_error_msg', None)), self)
apache_beam.runners.dataflow.dataflow_runner.DataflowRuntimeException: Dataflow pipeline failed. State: FAILED, Error:
Traceback (most recent call last):
  File "/usr/local/lib/python2.7/dist-packages/dataflow_worker/batchworker.py", line 773, in run
    self._load_main_session(self.local_staging_directory)
  File "/usr/local/lib/python2.7/dist-packages/dataflow_worker/batchworker.py", line 489, in _load_main_session
    pickler.load_session(session_file)
  File "/usr/local/lib/python2.7/dist-packages/apache_beam/internal/pickler.py", line 269, in load_session
    return dill.load_session(file_path)
  File "/usr/local/lib/python2.7/dist-packages/dill/_dill.py", line 410, in load_session
    module = unpickler.load()
  File "/usr/lib/python2.7/pickle.py", line 864, in load
    dispatch[key](self)
  File "/usr/lib/python2.7/pickle.py", line 1139, in load_reduce
    value = func(*args)
  File "/usr/local/lib/python2.7/dist-packages/dill/_dill.py", line 828, in _import_module
    return getattr(__import__(module, None, None, [obj]), obj)
  File "/usr/local/lib/python2.7/dist-packages/trainer/taxi.py", line 19, in <module>
    from tensorflow_transform import coders as tft_coders
  File "/usr/local/lib/python2.7/dist-packages/tensorflow_transform/__init__.py", line 19, in <module>
    from tensorflow_transform.analyzers import *
  File "/usr/local/lib/python2.7/dist-packages/tensorflow_transform/analyzers.py", line 39, in <module>
    from tensorflow_transform import tf_utils
  File "/usr/local/lib/python2.7/dist-packages/tensorflow_transform/tf_utils.py", line 24, in <module>
    from tensorflow.contrib.proto.python.ops import encode_proto_op
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/__init__.py", line 48, in <module>
    from tensorflow.contrib import distribute
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/distribute/__init__.py", line 34, in <module>
    from tensorflow.contrib.distribute.python.tpu_strategy import TPUStrategy
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/distribute/python/tpu_strategy.py", line 27, in <module>
    from tensorflow.contrib.tpu.python.ops import tpu_ops
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/tpu/__init__.py", line 73, in <module>
    from tensorflow.contrib.tpu.python.tpu.keras_support import tpu_model as keras_to_tpu_model
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/contrib/tpu/python/tpu/keras_support.py", line 71, in <module>
    from tensorflow.python.estimator import model_fn as model_fn_lib
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/estimator/__init__.py", line 25, in <module>
    import tensorflow.python.estimator.estimator_lib
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/estimator/estimator_lib.py", line 22, in <module>
    from tensorflow.python.estimator.canned.baseline import BaselineClassifier
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/estimator/canned/baseline.py", line 50, in <module>
    from tensorflow.python.estimator import estimator
ImportError: cannot import name estimator

python tfdv_analyze_and_validate.py \
  --input bigquery-public-data.chicago_taxi_trips.taxi_trips \
  --infer_schema \
  --stats_path ${TFDV_OUTPUT_PATH}/train_stats.tfrecord \
  --schema_path ${SCHEMA_PATH} \
  --project ${MYPROJECT} \
  --region us-central1 \
  --temp_location ${TEMP_PATH} \
  --experiments shuffle_mode=auto \
  --job_name ${JOB_ID} \
  --setup_file ./setup.py \
  --save_main_session \
  --runner DataflowRunner \
  --max_rows=${MAX_ROWS}

import setuptools

# LINT.IfChange
TF_VERSION = '1.13.1'
# LINT.ThenChange(train_mlengine.sh, start_model_server_mlengine.sh)

# LINT.IfChange
BEAM_VERSION = '2.12.0'
# LINT.ThenChange(setup_beam_on_flink.sh)

if __name__ == '__main__':
  setuptools.setup(
      name='tfx_chicago_taxi',
      version='0.13.0',
      packages=setuptools.find_packages(),
      install_requires=[
          'apache-beam[gcp]==' + BEAM_VERSION,
          'jupyter==1.0',
          'protobuf==3.7.1',
          'tensorflow==' + TF_VERSION,
          'tensorflow-data-validation==0.13.1',
          'tensorflow-metadata==0.13.0',
          'tensorflow-model-analysis==0.13.2',
          'tensorflow-serving-api==1.13.0',
          'tensorflow-transform==0.13.0',
      ],
      python_requires='>=2.7,<3')