update sample tfx taxi to meet PR kubeflow#823 changes

jinchihe · Feb 21, 2019 · 33a8faf · 33a8faf
1 parent e282bd3
commit 33a8faf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/samples/notebooks/KubeFlow Pipeline Using TFX OSS Components.ipynb b/samples/notebooks/KubeFlow Pipeline Using TFX OSS Components.ipynb
@@ -238,7 +238,7 @@
     "        name = step_name,\n",
     "        image = KUBEFLOW_DEPLOYER_IMAGE,\n",
     "        arguments = [\n",
-    "            '--model-path', model,\n",
+    "            '--model-export-path', '%s/export/export' % model,\n",
     "            '--server-name', tf_server_name\n",
     "        ]\n",
     "    )\n",

diff --git a/samples/tfx/taxi-cab-classification-pipeline.py b/samples/tfx/taxi-cab-classification-pipeline.py
@@ -132,7 +132,7 @@ def kubeflow_deploy_op(model: 'TensorFlow model', tf_server_name, step_name='dep
         name = step_name,
         image = 'gcr.io/ml-pipeline/ml-pipeline-kubeflow-deployer:5df2cdc1ed145320204e8bc73b59cdbd7b3da28f',
         arguments = [
-            '--model-path', model,
+            '--model-export-path', '%s/export/export' % model,
             '--server-name', tf_server_name
         ]
     )