changes based on kubeflow upgradation to list experiments

Issue-id: AIMLFW-102

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

diff --git a/kfadapter/kfadapter_main.py b/kfadapter/kfadapter_main.py
index 5376b2c..8126c59 100644 (file)
--- 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 (file)
--- 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")