From 16e156b5b47eac3422a026b4fc9185425cbe6d58 Mon Sep 17 00:00:00 2001
From: teddy <teddy.kim.kim@tf.dabeeo.com>
Date: Thu, 12 Feb 2026 17:52:42 +0900
Subject: [PATCH] =?UTF-8?q?=EB=AC=B8=EC=A0=9C=EB=90=98=EB=8A=94=20?=
 =?UTF-8?q?=ED=95=98=EC=9D=B4=ED=8D=BC=ED=8C=8C=EB=9D=BC=EB=AF=B8=ED=84=B0?=
 =?UTF-8?q?=20=EC=A3=BC=EC=84=9D=EC=B2=98=EB=A6=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../cd/training/model/service/ModelTrainMngService.java     | 5 ++---
 .../kamco/cd/training/model/service/TmpDatasetService.java  | 4 ++--
 .../kamco/cd/training/train/service/DockerTrainService.java | 6 +++---
 .../kamco/cd/training/train/service/TrainJobService.java    | 4 ++--
 4 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java b/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java
index 1db68cf..9567a5c 100644
--- a/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java
+++ b/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java
@@ -13,7 +13,6 @@ import com.kamco.cd.training.model.dto.ModelTrainMngDto.SearchReq;
 import com.kamco.cd.training.postgres.core.HyperParamCoreService;
 import com.kamco.cd.training.postgres.core.ModelTrainMngCoreService;
 import java.io.IOException;
-import java.nio.file.Path;
 import java.util.List;
 import java.util.UUID;
 import lombok.RequiredArgsConstructor;
@@ -102,9 +101,9 @@ public class ModelTrainMngService {
 
     try {
       // 데이터셋 심볼링크 생성
-      Path path = tmpDatasetService.buildTmpDatasetSymlink(raw, uids);
+      String tmpUid = tmpDatasetService.buildTmpDatasetSymlink(raw, uids);
       ModelTrainMngDto.UpdateReq updateReq = new ModelTrainMngDto.UpdateReq();
-      updateReq.setRequestPath(path.toString());
+      updateReq.setRequestPath(tmpUid);
       modelTrainMngCoreService.updateModelMaster(modelId, updateReq);
     } catch (IOException e) {
       throw new RuntimeException(e);
diff --git a/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java b/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java
index 04a969f..7f04c98 100644
--- a/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java
+++ b/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java
@@ -16,7 +16,7 @@ public class TmpDatasetService {
   @Value("${train.docker.requestDir}")
   private String requestDir;
 
-  public Path buildTmpDatasetSymlink(String uid, List<String> datasetUids) throws IOException {
+  public String buildTmpDatasetSymlink(String uid, List<String> datasetUids) throws IOException {
 
     log.info("========== buildTmpDatasetSymlink START ==========");
     log.info("uid={}", uid);
@@ -97,7 +97,7 @@ public class TmpDatasetService {
         regularFiles,
         symlinksMade);
 
-    return tmp;
+    return uid;
   }
 
   private static Path toPath(String p) {
diff --git a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
index d998df9..f56fcea 100644
--- a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
+++ b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
@@ -264,12 +264,12 @@ public class DockerTrainService {
 
     // ===== Augmentation =====
     addArg(c, "--rot-prob", req.getRotProb());
-    addArg(c, "--rot-degree", req.getRotDegree());
+    // addArg(c, "--rot-degree", req.getRotDegree()); // TODO AI 수정되면 주석 해제
     addArg(c, "--flip-prob", req.getFlipProb());
     addArg(c, "--exchange-prob", req.getExchangeProb());
     addArg(c, "--brightness-delta", req.getBrightnessDelta());
-    addArg(c, "--contrast-range", req.getContrastRange());
-    addArg(c, "--saturation-range", req.getSaturationRange());
+    // addArg(c, "--contrast-range", req.getContrastRange()); // TODO AI 수정되면 주석 해제
+    // addArg(c, "--saturation-range", req.getSaturationRange()); // TODO AI 수정되면 주석 해제
     addArg(c, "--hue-delta", req.getHueDelta());
 
     addArg(c, "--resume-from", req.getResumeFrom());
diff --git a/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java b/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java
index 0acad20..d3f11c7 100644
--- a/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java
+++ b/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java
@@ -226,9 +226,9 @@ public class TrainJobService {
 
     try {
       // 데이터셋 심볼링크 생성
-      Path path = tmpDatasetService.buildTmpDatasetSymlink(raw, uids);
+      String pathUid = tmpDatasetService.buildTmpDatasetSymlink(raw, uids);
       ModelTrainMngDto.UpdateReq updateReq = new ModelTrainMngDto.UpdateReq();
-      updateReq.setRequestPath(path.toString());
+      updateReq.setRequestPath(pathUid);
       modelTrainMngCoreService.updateModelMaster(modelId, updateReq);
     } catch (IOException e) {
       throw new RuntimeException(e);