[Sample] Update pre-load TFX::OSS sample

backend/src/apiserver/config/sample_config.json

@@ -25,7 +25,7 @@
     "file": "/samples/core/exit_handler/exit_handler.py.tar.gz"
   },
   {
-    "name": "[Sample] ML - TFX - Taxi Tip Prediction Model Trainer",
+    "name": "[Sample] Unified DSL - Taxi Tip Prediction Model Trainer",
     "description": "[GCP Permission requirements](https://github.com/kubeflow/pipelines/blob/master/samples/contrib/parameterized_tfx_oss#permission). Example pipeline that does classification with model analysis based on a public tax cab BigQuery dataset. [source code](https://github.com/kubeflow/pipelines/tree/master/samples/contrib/parameterized_tfx_oss)",
     "file": "/samples/contrib/parameterized_tfx_oss/parameterized_tfx_oss.tar.gz"
   }

samples/contrib/parameterized_tfx_oss/README.md

@@ -9,11 +9,13 @@ This sample demonstrates how to author a ML pipeline in TFX and run it on a KFP
 Please refer to inline comments for the purpose of each step.
 
 In order to successfully compile this sample, you'll need to have a TFX installation at HEAD.
-First, you can clone their repo and
-then point the version tag in `tfx/version.py` to TFX's latest nightly build image of version 
-`0.15.0dev` (e.g., `0.15.0dev20191007`, list of image available can be found [here](https://hub.docker.com/r/tensorflow/tfx/tags)).
-Finally, run `python setup.py install` from `tfx/tfx`. After that, running 
-`chicago_taxi_pipeline_simple.py` compiles the TFX pipeline into KFP pipeline package.
+First, you can clone their repo and run `python setup.py install` from `tfx/tfx`. 
+The image used in the pipeline is specified as `tfx_image` in the 
+`KubeflowDagRunnerConfig`. Currently we're using our own patched version of TFX image containing visualization support.
+List of officially released nightly build image available can be found [here](https://hub.docker.com/r/tensorflow/tfx/tags)).
+
+After that, running 
+`python3 chicago_taxi_pipeline_simple.py` compiles the TFX pipeline into KFP pipeline package.
 This pipeline requires google storage permission to run. 
 
 

samples/contrib/parameterized_tfx_oss/parameterized_tfx_oss.py

@@ -141,7 +141,8 @@ def _get_kubeflow_metadata_config() -> kubeflow_pb2.KubeflowMetadataConfig:
       enable_cache=enable_cache)
 
   config = kubeflow_dag_runner.KubeflowDagRunnerConfig(
-      kubeflow_metadata_config=_get_kubeflow_metadata_config())
+      kubeflow_metadata_config=_get_kubeflow_metadata_config(),
+      tfx_image='gcr.io/ml-pipeline/patched-tfx:0.1.32')
 
   kfp_runner = kubeflow_dag_runner.KubeflowDagRunner(config=config)
   # Make sure kfp_runner recognizes those parameters.