From: rajdeep11 <rajdeep.sin@samsung.com>
Date: Wed, 26 Jun 2024 23:43:00 +0000 (+0530)
Subject: changes for the run pipeline
X-Git-Tag: 3.0.0~11
X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=commitdiff_plain;h=f051dedf9a845115fc5a1edef348ce0c6fbf1e61;p=aiml-fw%2Fathp%2Ftps%2Fkubeflow-adapter.git

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
--- 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
--- 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: