From c54f1ea865f0a8f55997f030a152bdcd3c992fef Mon Sep 17 00:00:00 2001 From: rajdeep11 Date: Thu, 27 Jun 2024 05:03:04 +0530 Subject: [PATCH] based on kubeflow upgradation Issue-id: AIMLFW-102 Change-Id: I7321e226446f8b764254e568f521ad18fd465faa Signed-off-by: rajdeep11 --- kfadapter/kfadapter_util.py | 2 +- test/test_kfadapter_util.py | 10 +++++----- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/kfadapter/kfadapter_util.py b/kfadapter/kfadapter_util.py index 4b52ddc..8fb230e 100644 --- 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 --- 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): -- 2.16.6