Handle NO_CATCH issue

Added missing try-catch statement
resolve 457589

Signed-off-by: hyeonseok lee <hs89.lee@samsung.com>

diff --git a/Applications/SimpleShot/task_runner.cpp b/Applications/SimpleShot/task_runner.cpp
index be2c0ad20f9d9a4b439eb58f8a84959092ca6156..6128a6480297956851b23f0d7b6942cd04ed7639 100644 (file)
--- a/Applications/SimpleShot/task_runner.cpp
+++ b/Applications/SimpleShot/task_runner.cpp
@@ -246,10 +246,16 @@ int main(int argc, char **argv) {
     return 1;
   }
 
-  std::shared_ptr<ml::train::Dataset> train_dataset = ml::train::createDataset(
-    ml::train::DatasetType::FILE,
-    {"train_data=" + train_path, "val_data=" + val_path,
-     "label_data=" + label_path});
+  std::shared_ptr<ml::train::Dataset> train_dataset;
+  try {
+    train_dataset = ml::train::createDataset(ml::train::DatasetType::FILE,
+                                             {"train_data=" + train_path,
+                                              "val_data=" + val_path,
+                                              "label_data=" + label_path});
+  } catch (...) {
+    std::cerr << "creating dataset failed";
+    return 1;
+  }
 
   if (model->setDataset(train_dataset)) {
     std::cerr << "failed to set dataset" << std::endl;