From bc626950e0fa4cdcf9af7ce75d39cc628af603e6 Mon Sep 17 00:00:00 2001
From: ashishj1729 <jain.ashish@samsung.com>
Date: Tue, 3 Jun 2025 17:37:47 +0530
Subject: [PATCH] Updating Pipeline as per new MM-SDK upload_metrics and
 get_model_metrics function definations Issue-Id: AIMLFW-209

Change-Id: I4d95c2afefc67e5364020380d1a361fb91b8d1b4
Signed-off-by: ashishj1729 <jain.ashish@samsung.com>
---
 kf-pipelines/qoe-pipeline.ipynb | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/kf-pipelines/qoe-pipeline.ipynb b/kf-pipelines/qoe-pipeline.ipynb
index f688616..edaf838 100644
--- a/kf-pipelines/qoe-pipeline.ipynb
+++ b/kf-pipelines/qoe-pipeline.ipynb
@@ -24,7 +24,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -110,13 +110,17 @@
     "    updated_model_info= requests.post(url).json()\n",
     "    print(updated_model_info)\n",
     "    \n",
-    "    mm_sdk.upload_metrics(data, modelname, modelversion,artifactversion)\n",
-    "    mm_sdk.upload_model(\"./\", modelname, modelversion, artifactversion)"
+    "    #featurepath is a combination of <feature_group>_<trainingjob_Id>\n",
+    "    trainingjob_id = featurepath.split('_')[-1]\n",
+    "    mm_sdk.upload_metrics(data, trainingjob_id)\n",
+    "    print(\"Model-metric : \", mm_sdk.get_metrics(trainingjob_id))\n",
+    "    mm_sdk.upload_model(\"./\", modelname, modelversion, artifactversion)\n",
+    "    "
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 4,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -134,7 +138,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 5,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -147,7 +151,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 6,
    "metadata": {},
    "outputs": [
     {
@@ -156,7 +160,7 @@
        "<Response [200]>"
       ]
      },
-     "execution_count": 7,
+     "execution_count": 6,
      "metadata": {},
      "output_type": "execute_result"
     }
-- 
2.16.6