From: rajdeep11 Date: Wed, 26 Jun 2024 22:12:43 +0000 (+0530) Subject: changes based on kubeflow upgradation to list experiments X-Git-Tag: 3.0.0~16 X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=commitdiff_plain;h=refs%2Fchanges%2F62%2F13062%2F1;p=aiml-fw%2Fathp%2Ftps%2Fkubeflow-adapter.git changes based on kubeflow upgradation to list experiments Issue-id: AIMLFW-102 Change-Id: I12aa62f5dd83dbc0cd16ae329a6a7b2390f9cc61 Signed-off-by: rajdeep11 --- diff --git a/kfadapter/kfadapter_main.py b/kfadapter/kfadapter_main.py index 5376b2c..8126c59 100644 --- a/kfadapter/kfadapter_main.py +++ b/kfadapter/kfadapter_main.py @@ -277,7 +277,7 @@ def list_experiments(): try: exp = KFCONNECT_KF_OBJ.get_kf_list_experiments(KFCONNECT_CONFIG_OBJ.kf_dict['kfdefaultns']) for experiment in exp.experiments: - exp_dict[experiment.name] = experiment.id + exp_dict[experiment.display_name] = experiment.experiment_id except:# pylint: disable=bare-except tbk = traceback.format_exc() LOGGER.error(tbk) diff --git a/test/test_kfadapter_main.py b/test/test_kfadapter_main.py index 3a1afc0..ca3602a 100644 --- a/test/test_kfadapter_main.py +++ b/test/test_kfadapter_main.py @@ -275,8 +275,8 @@ class testKfadapterApi(TestCase): def test_get_experiments(self, mock_get_kf_list_experiments): # given exp = kfp_server_api.ApiExperiment() - exp.name = "exp-name" - exp.id = "exp-id" + exp.display_name = "exp-name" + exp.experiment_id = "exp-id" explist = kfp_server_api.ApiListExperimentsResponse() explist.experiments = [exp] @@ -289,7 +289,7 @@ class testKfadapterApi(TestCase): mock_get_kf_list_experiments.assert_called_once() self.assertEqual(response.content_type, "application/json") self.assertEqual(response.status_code, status.HTTP_200_OK) - self.assertEqual(response.get_json()[exp.name], exp.id) + self.assertEqual(response.get_json()[exp.display_name], exp.experiment_id) @patch("kfadapter.kfadapter_kfconnect.KfConnect.get_kf_experiment_details")