based on kubeflow upgradation

 Issue-id: AIMLFW-102

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

diff --git a/kfadapter/kfadapter_util.py b/kfadapter/kfadapter_util.py
index 4b52ddc..8fb230e 100644 (file)
--- a/kfadapter/kfadapter_util.py
+++ b/kfadapter/kfadapter_util.py
@@ -77,7 +77,7 @@ def run_finished(run_status: string) -> bool:
             true or false to signify whether run is finished or not
 
     """
-    return run_status in {'Succeeded', 'Failed', 'Error', 'Skipped', 'Terminated'}
+    return run_status in {'SUCCEEDED', 'FAILED', 'ERROR', 'SKIPPED', 'TERMINATED'}
 
 def random_suffix():
     """

diff --git a/test/test_kfadapter_util.py b/test/test_kfadapter_util.py
index 0e3618e..7d7683a 100644 (file)
--- a/test/test_kfadapter_util.py
+++ b/test/test_kfadapter_util.py
@@ -40,23 +40,23 @@ class Test_kfadapter_util:
         assert str(ret).islower() == True
         
     def test_run_finished_succeeded(self):
-        ret = kfadapter_util.run_finished('Succeeded')
+        ret = kfadapter_util.run_finished('SUCCEEDED')
         assert ret == True
     
     def test_run_finished_failed(self):
-        ret = kfadapter_util.run_finished('Failed')
+        ret = kfadapter_util.run_finished('FAILED')
         assert ret == True
         
     def test_run_finished_error(self):
-        ret = kfadapter_util.run_finished('Error')
+        ret = kfadapter_util.run_finished('ERROR')
         assert ret == True
         
     def test_run_finished_skipped(self):
-        ret = kfadapter_util.run_finished('Skipped')
+        ret = kfadapter_util.run_finished('SKIPPED')
         assert ret == True
         
     def test_run_finished_terminated(self):
-        ret = kfadapter_util.run_finished('Terminated')
+        ret = kfadapter_util.run_finished('TERMINATED')
         assert ret == True
 
     def test_check_list(self):