fix bug in the notebook sample (#419)

kubeflow · Nov 30, 2018 · 052de19 · 052de19
1 parent 9a62c31
commit 052de19
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/samples/notebooks/KubeFlow Pipeline Using TFX OSS Components.ipynb b/samples/notebooks/KubeFlow Pipeline Using TFX OSS Components.ipynb
@@ -548,6 +548,7 @@
     "    component_func=deploy_model,\n",
     "    staging_gcs_path=OUTPUT_DIR,\n",
     "    dependency=[kfp.compiler.VersionedDependency(name='google-api-python-client', version='1.7.0')],\n",
+    "    base_image='tensorflow/tensorflow:1.12.0',\n",
     "    target_image=TARGET_IMAGE)"
    ]
   },