changes based on kubeflow upgradation

Issue-id: AIMLFW-86

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

diff --git a/tests/test_tm_apis.py b/tests/test_tm_apis.py
index bef6bff..8a80de8 100644 (file)
--- 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 (file)
--- 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)