do not use ml_train_dataset_create_with_generator function

ml_train_dataset_create_with_generator function has been deprecated
according to https://github.com/nnstreamer/nntrainer/blob/main/api/capi/include/nntrainer.h#L418
Use ml_train_dataset_create function instead of it.

Change-Id: I233631220cab8f085ae2a9e8bb3d5b18cf7f8c0e
Signed-off-by: Inki Dae <inki.dae@samsung.com>

diff --git a/src/training_engine_nntrainer.cpp b/src/training_engine_nntrainer.cpp
index df958a7b73c54d509aca59fc1a755228d5afa2f6..6d38e37d016e0d774523a84280215ad6ecf62976 100644 (file)
--- a/src/training_engine_nntrainer.cpp
+++ b/src/training_engine_nntrainer.cpp
@@ -309,23 +309,7 @@ namespace NntrainerImpl
                ml_train_dataset_h handle;
                int ret = ML_ERROR_NONE;
 
-               switch (type) {
-                       case TRAINING_DATASET_TYPE_TRAIN:
-                               ret = ml_train_dataset_create_with_generator(&handle, UpdateTrainData, NULL, NULL);
-                               break;
-                       case TRAINING_DATASET_TYPE_TEST:
-                               ret = ml_train_dataset_create_with_generator(&handle, NULL, NULL, UpdateTrainData);
-                               break;
-                       case TRAINING_DATASET_TYPE_VERIFY:
-                               ret = ml_train_dataset_create_with_generator(&handle, NULL, UpdateTrainData, NULL);
-                               break;
-                       default:
-                               LOGE("Invalid dataset type.");
-                               return NULL;
-               }
-
-               LOGI("LEAVE");
-
+               ret = ml_train_dataset_create(&handle);
                if (ret != ML_ERROR_NONE) {
                        LOGE("Failed to create a dataset.");
                        return NULL;
@@ -335,6 +319,8 @@ namespace NntrainerImpl
                dataset->type = type;
                dataset->dataset_handle = static_cast<void *>(handle);
 
+               LOGI("LEAVE");
+
                return dataset;
        }