add importing yaml library (kubeflow#553)

kwonmha · Mar 25, 2021 · a45c49f · a45c49f
1 parent e62a6c8
commit a45c49f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/mnist/mnist_e2e_on_prem.ipynb b/examples/mnist/mnist_e2e_on_prem.ipynb
@@ -132,6 +132,7 @@
     "from kubernetes import client as k8s_client\n",
     "from kubernetes import config as k8s_config\n",
     "from kubeflow.fairing.utils import is_running_in_k8s\n",
+    "import yaml",
     "\n",
     "pv_yaml = f'''\n",
     "apiVersion: v1\n",