changes for the run pipeline

Issue-id: AIMLFW-102

Change-Id: I7b090776dfa28709bba3adbef9d16e6d7639125d
Signed-off-by: rajdeep11 <rajdeep.sin@samsung.com>

diff --git a/kfadapter/kfadapter_kfconnect.py b/kfadapter/kfadapter_kfconnect.py
index 11b0b1f..eae4fcc 100644 (file)
--- a/kfadapter/kfadapter_kfconnect.py
+++ b/kfadapter/kfadapter_kfconnect.py
@@ -262,10 +262,20 @@ class KfConnect:
 
         """
         self.logger.debug("run_kf_pipeline Entered")
-        run = self.kfp_client.run_pipeline(exp_id, job_name="testjob_"+\
+        job_name=arguments["trainingjob_name"]
+        req_dict={
+            "trainingjob_name":job_name,
+            "epochs": arguments["epochs"],
+            "version": str(arguments["version"])
+        }
+        try:
+            run = self.kfp_client.run_pipeline(exp_id, job_name="testjob"+\
                                            random_suffix(),
-                                           pipeline_package_path=None, params=arguments,
+                                           pipeline_package_path=None, params=req_dict,
                                            pipeline_id=pipeline_id,
                                            version_id=version_id)
+        except Exception as err:
+            self.logger.error(str(err))
+
         self.logger.debug("run_kf_pipeline Exited")
         return run
\ No newline at end of file

diff --git a/test/test_kfadapter_kfconnect.py b/test/test_kfadapter_kfconnect.py
index d15b8d1..b1971a0 100644 (file)
--- a/test/test_kfadapter_kfconnect.py
+++ b/test/test_kfadapter_kfconnect.py
@@ -82,7 +82,11 @@ class Test_KfConnect:
         assert None == self.__KFCONNECT.delete_kf_pipeline(pipeline_id='pipeline_id')
 
     def test_run_kf_pipeline(self):
-        assert None == self.__KFCONNECT.run_kf_pipeline(exp_id='exp_id', pipeline_id='pipeline_id', arguments='arguments', version_id='version_id')
+        assert None == self.__KFCONNECT.run_kf_pipeline(exp_id='exp_id', pipeline_id='pipeline_id', arguments={
+            "trainingjob_name":"job_name",
+            "epochs": "epochs",
+            "version": "version"}
+        , version_id='version_id')
     
 
 class Test_Negative_KfConnect: