From: Inki Dae <inki.dae@samsung.com>
Date: Fri, 20 Aug 2021 08:39:02 +0000 (+0900)
Subject: do not use ml_train_dataset_create_with_generator function
X-Git-Tag: submit/tizen/20220307.024724~9
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=4ce404a79bf7bdc7e61201c7c8f7069f3c4af6d0;p=platform%2Fupstream%2Ftraining-engine-nntrainer.git

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 df958a7..6d38e37 100644
--- 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;
 	}