changes in the training function

updating the imports

Change-Id: I36b17c9c9badcde5fb5a4e670a1cc38bdf949ba5
Signed-off-by: rajdeep11 <rajdeep.sin@samsung.com>
(cherry picked from commit 14bdf18ee72b4bc9d2757b1bc11265095947e6ee)

diff --git a/kf-pipelines/qoe-pipeline.ipynb b/kf-pipelines/qoe-pipeline.ipynb
index 7bd372d..1cb2c88 100644 (file)
--- a/kf-pipelines/qoe-pipeline.ipynb
+++ b/kf-pipelines/qoe-pipeline.ipynb
@@ -9,7 +9,8 @@
     "import kfp\n",
     "import kfp.dsl as dsl\n",
     "from kfp.dsl import InputPath, OutputPath\n",
-    "from kfp.dsl import component as component"
+    "from kfp.dsl import component as component\n",
+    "from kfp import kubernetes"
    ]
   },
   {