모델등록 수정

2026-02-12 15:43:52 +09:00
parent 7a22d8ba73
commit 8e4bea53da
3 changed files with 15 additions and 11 deletions
--- a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
+++ b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java
@@ -78,7 +78,7 @@ public class ModelTrainMngCoreService {
   * @param addReq
   * @return
   */
-  public Long saveModel(ModelTrainMngDto.AddReq addReq) {
+  public ModelTrainMngDto.Basic saveModel(ModelTrainMngDto.AddReq addReq) {
    ModelMasterEntity entity = new ModelMasterEntity();
    ModelHyperParamEntity hyperParamEntity = new ModelHyperParamEntity();

@@ -117,7 +117,12 @@ public class ModelTrainMngCoreService {

    entity.setCreatedUid(userUtil.getId());
    ModelMasterEntity resultEntity = modelMngRepository.save(entity);
-    return resultEntity.getId();
+
+    ModelTrainMngDto.Basic result = new ModelTrainMngDto.Basic();
+    result.setId(resultEntity.getId());
+    result.setUuid(resultEntity.getUuid());
+
+    return result;
  }

  /**