From b8fe1a1895b38fcc910678e48d000f3a28dfc387 Mon Sep 17 00:00:00 2001 From: rajdeep11 Date: Wed, 26 Jun 2024 23:57:35 +0530 Subject: [PATCH] changes based on kubeflow upgradation Issue-id: AIMLFW-86 Change-Id: Ie5093097ab31f7d77eaa1e096a078f60c9e902f3 Signed-off-by: rajdeep11 --- tests/test_tm_apis.py | 6 +++--- trainingmgr/trainingmgr_main.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_tm_apis.py b/tests/test_tm_apis.py index bef6bff..8a80de8 100644 --- a/tests/test_tm_apis.py +++ b/tests/test_tm_apis.py @@ -154,7 +154,7 @@ class Test_pipeline_notification: trainingmgr_main.LOGGER.debug("******* test_pipeline_notification post *******") trainingjob_req = { "trainingjob_name":"usecase1", - "run_status":"Succeeded", + "run_status":"SUCCEEDED", } expected_data = "Pipeline notification success." response = self.client.post("/trainingjob/pipelineNotification".format("usecase1"),data=json.dumps(trainingjob_req), @@ -181,7 +181,7 @@ class Test_pipeline_notification: trainingmgr_main.LOGGER.debug("******* test_pipeline_notification post *******") trainingjob_req = { "trainingjob_name":"usecase1", - "run_status":"Succeeded", + "run_status":"SUCCEEDED", } expected_data = "Pipeline notification success." response = self.client.post("/trainingjob/pipelineNotification".format("usecase1"),data=json.dumps(trainingjob_req), @@ -208,7 +208,7 @@ class Test_pipeline_notification: trainingmgr_main.LOGGER.debug("******* test_pipeline_notification post *******") trainingjob_req = { "trainingjob_name":"usecase1", - "run_status":"Succeeded", + "run_status":"SUCCEEDED", } try: response = self.client.post("/trainingjob/pipelineNotification".format("usecase1"),data=json.dumps(trainingjob_req), diff --git a/trainingmgr/trainingmgr_main.py b/trainingmgr/trainingmgr_main.py index 993b652..b45442a 100644 --- a/trainingmgr/trainingmgr_main.py +++ b/trainingmgr/trainingmgr_main.py @@ -502,7 +502,7 @@ def pipeline_notification(): trainingjob_name = request.json["trainingjob_name"] run_status = request.json["run_status"] - if run_status == 'Succeeded': + if run_status == 'SUCCEEDED': change_steps_state_of_latest_version(trainingjob_name, PS_DB_OBJ, Steps.TRAINING.name, States.FINISHED.name) -- 2.16.6