[Visualization] tfdv visualization throws an error

[Bobgy]

What happened:

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-114-d62b8c370a9d> in <module>
----> 1 import tensorflow_data_validation as tfdv
      2 stats = tfdv.load_statistics('gs://hostedkfp-default-0p59u0vc8g/tfx_taxi_simple/88332a30-3d2a-4c1e-a9e7-f35f972523ff/StatisticsGen/statistics/2/eval/stats_tfrecord')
      3 tfdv.visualize_statistics(stats)

/usr/local/lib/python3.6/dist-packages/tensorflow_data_validation/__init__.py in <module>
     22 
     23 # Import stats API.
---> 24 from tensorflow_data_validation.api.stats_api import GenerateStatistics
     25 
     26 # Import validation API.

/usr/local/lib/python3.6/dist-packages/tensorflow_data_validation/api/stats_api.py in <module>
     50 from tensorflow_data_validation import constants
     51 from tensorflow_data_validation import types
---> 52 from tensorflow_data_validation.statistics import stats_impl
     53 from tensorflow_data_validation.statistics import stats_options
     54 from typing import Generator

/usr/local/lib/python3.6/dist-packages/tensorflow_data_validation/statistics/stats_impl.py in <module>
     31 from tensorflow_data_validation import constants
     32 from tensorflow_data_validation import types
---> 33 from tensorflow_data_validation.arrow import arrow_util
     34 from tensorflow_data_validation.statistics import stats_options
     35 from tensorflow_data_validation.statistics.generators import basic_stats_generator

/usr/local/lib/python3.6/dist-packages/tensorflow_data_validation/arrow/arrow_util.py in <module>
     22 import pyarrow as pa
     23 from tensorflow_data_validation import types
---> 24 from tfx_bsl.arrow import array_util
     25 from typing import Iterable, Optional, Text, Tuple
     26 

/usr/local/lib/python3.6/dist-packages/tfx_bsl/arrow/array_util.py in <module>
     15 # pytype: disable=import-error
     16 # pylint: disable=wildcard-import
---> 17 from tfx_bsl.cc.tfx_bsl_extension.arrow.array_util import *
     18 # pytype: enable=import-error
     19 # pylint: enable=wildcard-import

ImportError: libarrow.so.15: cannot open shared object file: No such file or directory

After #3060, tfdv throws an error.

What did you expect to happen:
It should show visualization.

What steps did you take:
[A clear and concise description of what the bug is.]

1. use kfp 0.2.2
2. run tfx sample
3. view statisticsgen step
4. the tab shows broken visualization

Anything else you would like to add:
[Miscellaneous information that will assist in solving the issue.]

[Bobgy]

/assign @jingzhang36
/kind bug
/priority p0

[jingzhang36]

After we bump tfdv version to 0.21.1, our visualization server seems to have a pyarrow at 0.16.x. We'll have to pin it to 0.15.0

[Bobgy]

That seems another failure point to let us prioritize #3078

[Bobgy]

Thanks for the quick check!

[rmgogogo]

+rmgogogo
we may provide a quick patch after the fix (only patch visualization server)