Merge pull request tensorflow#1279 from dhruvesh09/master

Update Version
TheMichaelHu · Feb 13, 2020 · 8f5a13f · 8f5a13f
2 parents 392649c + 45f4785
commit 8f5a13f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/docs/tutorials/tfx/template.ipynb b/docs/tutorials/tfx/template.ipynb
@@ -205,7 +205,8 @@
     "# This refers to the KFP cluster endpoint\n",
     "ENDPOINT='' # Enter your ENDPOINT here.\n",
     "if not ENDPOINT:\n",
-    "    raise Exception('Set your ENDPOINT in this cell.')"
+    "    from absl import logging\n",
+    "    logging.error('Set your ENDPOINT in this cell.')"
    ]
   },
   {
@@ -550,7 +551,8 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "However, we recommend visiting your KFP Dashboard using Web Browser. If you launched your KFP cluster in GCP, you can access KFP Dashboard from the Cloud AI Platform Pipelines menu in Google Cloud Console. Once you visit the dashboard, you will be able to find the pipeline, the run and many more information about the pipeline."
+    "However, we recommend visiting your KFP Dashboard using Web Browser. If you launched your KFP cluster in GCP, you can access KFP Dashboard from the Cloud AI Platform Pipelines menu in Google Cloud Console. Once you visit the dashboard, you will be able to find the pipeline, the run and many more information about the pipeline.\n",
+    "For example, you can find your runs under *Experiments* menu, and you can find all your artifacts from the pipeline under *Artifacts* menu.\n"
    ]
   },
   {

diff --git a/tfx/examples/iris/iris_pipeline_native_keras.py b/tfx/examples/iris/iris_pipeline_native_keras.py
@@ -101,6 +101,7 @@ def _create_pipeline(pipeline_name: Text, pipeline_root: Text, data_root: Text,
       examples=example_gen.outputs['examples'],
       model=trainer.outputs['model'],
       eval_config=tfma.EvalConfig(
+          model_specs=[tfma.ModelSpec(label_key='variety')],
           slicing_specs=[tfma.SlicingSpec(feature_keys=['sepal_length'])]))
 
   # TODO(jyzhao): support model validation in evaluator and pusher component.

diff --git a/tfx/extensions/google_cloud_ai_platform/runner.py b/tfx/extensions/google_cloud_ai_platform/runner.py
@@ -104,7 +104,7 @@ def start_aip_training(input_dict: Dict[Text, List[types.Artifact]],
     output_dict: Passthrough input dict for tfx.components.Trainer.executor.
     exec_properties: Passthrough input dict for tfx.components.Trainer.executor.
     executor_class_path: class path for TFX core default trainer.
-    training_inputs: Training input argment for AI Platform training job.
+    training_inputs: Training input argument for AI Platform training job.
       'pythonModule', 'pythonVersion' and 'runtimeVersion' will be inferred. For
       the full set of parameters, refer to
       https://cloud.google.com/ml-engine/reference/rest/v1/projects.jobs#TrainingInput