From f051dedf9a845115fc5a1edef348ce0c6fbf1e61 Mon Sep 17 00:00:00 2001 From: rajdeep11 Date: Thu, 27 Jun 2024 05:13:00 +0530 Subject: [PATCH] changes for the run pipeline Issue-id: AIMLFW-102 Change-Id: I7b090776dfa28709bba3adbef9d16e6d7639125d Signed-off-by: rajdeep11 --- kfadapter/kfadapter_kfconnect.py | 14 ++++++++++++-- test/test_kfadapter_kfconnect.py | 6 +++++- 2 files changed, 17 insertions(+), 3 deletions(-) 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: -- 2.16.6