[ML][Train] Unify thrown exceptions and some fixes.

[platform/core/api/webapi-plugins.git] / src / ml / ml_trainer_manager.cc

diff --git a/src/ml/ml_trainer_manager.cc b/src/ml/ml_trainer_manager.cc
index 22805e8..9b81d66 100644 (file)
--- a/src/ml/ml_trainer_manager.cc
+++ b/src/ml/ml_trainer_manager.cc
@@ -77,7 +77,7 @@ PlatformResult TrainerManager::ModelCompile(int id,
 
   if (models_.find(id) == models_.end()) {
     LoggerE("Could not find model with id: %d", id);
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find model");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find model");
   }
 
   auto& model = models_[id];
@@ -138,7 +138,7 @@ PlatformResult TrainerManager::ModelRun(int id,
   if (models_.find(id) == models_.end()) {
     LoggerE("Could not find model with id: %d", id);
     models_map_lock_.unlock();
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find model");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find model");
   }
 
   auto& model = models_[id];
@@ -195,7 +195,7 @@ PlatformResult TrainerManager::ModelAddLayer(int id, int layerId) {
 
   if (models_.find(id) == models_.end()) {
     LoggerE("Could not find model with id: %d", id);
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find model");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find model");
   }
 
   auto& model = models_[id];
@@ -209,7 +209,7 @@ PlatformResult TrainerManager::ModelAddLayer(int id, int layerId) {
   if (layers_.find(layerId) == layers_.end()) {
     LoggerE("Could not find layer with id: %d", id);
     model->instanceLock.unlock();
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find layer");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find layer");
   }
 
   auto& layer = layers_[layerId];
@@ -243,7 +243,7 @@ PlatformResult TrainerManager::ModelSetOptimizer(int id, int optimizerId) {
 
   if (models_.find(id) == models_.end()) {
     LoggerE("Could not find model with id: %d", id);
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find model");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find model");
   }
 
   auto& model = models_[id];
@@ -257,7 +257,7 @@ PlatformResult TrainerManager::ModelSetOptimizer(int id, int optimizerId) {
   if (optimizers_.find(optimizerId) == optimizers_.end()) {
     LoggerE("Could not find optimizer with id: %d", id);
     model->instanceLock.unlock();
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find optimizer");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find optimizer");
   }
 
   auto& optimizer = optimizers_[optimizerId];
@@ -318,7 +318,7 @@ PlatformResult TrainerManager::ModelSetDataset(int id, int datasetId) {
   if (datasets_.find(datasetId) == datasets_.end()) {
     LoggerE("Could not find dataset with id: %d", id);
     model->instanceLock.unlock();
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find dataset");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find dataset");
   }
 
   auto& dataset = datasets_[datasetId];
@@ -359,7 +359,7 @@ PlatformResult TrainerManager::ModelSummarize(int id,
 
   if (models_.find(id) == models_.end()) {
     LoggerE("Could not find model with id: %d", id);
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find model");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find model");
   }
 
   auto& model = models_[id];
@@ -455,7 +455,7 @@ PlatformResult TrainerManager::ModelSave(int id, const std::string& path,
 
   if (models_.find(id) == models_.end()) {
     LoggerE("Could not find model with id: %d", id);
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find model");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find model");
   }
 
   auto& model = models_[id];
@@ -493,7 +493,7 @@ PlatformResult TrainerManager::ModelLoad(int id, const std::string& path,
 
   if (models_.find(id) == models_.end()) {
     LoggerE("Could not find model with id: %d", id);
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find model");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find model");
   }
 
   auto& model = models_[id];
@@ -606,7 +606,7 @@ PlatformResult TrainerManager::LayerSetProperty(int id,
 
   if (layers_.find(id) == layers_.end()) {
     LoggerE("Could not find layer with id: %d", id);
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find layer");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find layer");
   }
 
   auto layer = layers_[id];
@@ -701,7 +701,7 @@ PlatformResult TrainerManager::OptimizerSetProperty(int id,
 
   if (optimizers_.find(id) == optimizers_.end()) {
     LoggerE("Could not find optimizer with id: %d", id);
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find optimizer");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find optimizer");
   }
 
   auto optimizer = optimizers_[id];
@@ -831,7 +831,7 @@ PlatformResult TrainerManager::DatasetSetProperty(
 
   if (datasets_.find(id) == datasets_.end()) {
     LoggerE("Could not find dataset with id: %d", id);
-    return PlatformResult(ErrorCode::ABORT_ERR, "Could not find dataset");
+    return PlatformResult(ErrorCode::NOT_FOUND_ERR, "Could not find dataset");
   }
 
   auto dataset = datasets_[id];