Merge pull request #45 from chamathpali/dev

AI Task Issue fix
isee4xai · Feb 10, 2023 · 2ab7caf · 2ab7caf
2 parents e6fcae5 + 019f68d
commit 2ab7caf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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();