From 019f68db2880019e0b241f3f678de81782ce9bca Mon Sep 17 00:00:00 2001
From: Chamath Palihawadana <chamathpali123@gmail.com>
Date: Fri, 10 Feb 2023 09:50:50 +0000
Subject: [PATCH] AI Task Issue fix

---
 src/controllers/usecase.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/controllers/usecase.js b/src/controllers/usecase.js
index e0527b0..32f0826 100644
--- a/src/controllers/usecase.js
+++ b/src/controllers/usecase.js
@@ -265,7 +265,7 @@ module.exports.updateModel = async (req, res) => {
     await dataset_file.mv(path_dataset)
 
     // FUTURE IMPROVEMENTS: Parse iSee Data to ML Lib Format
-    let ai_task = req.body.ai_task.split("#")[1];
+    let ai_task = req.body.ai_tasks[ai_tasks.length - 1].split("#")[1];
     let dataset_type = req.body.dataset_type.split("#")[1];
     let backend = req.body.backend.split("#")[1].toLowerCase();