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 3a82566..e7a6dc1 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 @@ -12,7 +12,6 @@ import com.kamco.cd.training.model.dto.ModelTrainMngDto; 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 com.kamco.cd.training.train.service.TrainJobService; import java.io.IOException; import java.nio.file.Path; import java.util.List; @@ -30,7 +29,6 @@ import org.springframework.transaction.annotation.Transactional; @Slf4j public class ModelTrainMngService { - private final TrainJobService trainJobService; private final ModelTrainMngCoreService modelTrainMngCoreService; private final HyperParamCoreService hyperParamCoreService; private final TmpDatasetService tmpDatasetService; 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 c58f135..e707f69 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 @@ -14,14 +14,14 @@ import org.springframework.transaction.annotation.Transactional; @RequiredArgsConstructor public class TmpDatasetService { - @Value("${train.requestDir}") + @Value("${train.docker.requestDir}") private String requestDir; - // 환경에 맞게 yml로 빼는 걸 추천 - private final Path BASE = Paths.get(requestDir); - @Transactional(readOnly = true) - public Path buildTmpDatasetSymlink(String uid, List uids) throws IOException { + public Path buildTmpDatasetSymlink(String uid, List datasetUids) throws IOException { + + // 환경에 맞게 yml로 빼는 걸 추천 + Path BASE = Paths.get(requestDir); Path tmp = BASE.resolve("tmp").resolve(uid); // mkdir -p "$TMP"/train/{input1,input2,label} ... @@ -31,7 +31,7 @@ public class TmpDatasetService { } } - for (String id : uids) { + for (String id : datasetUids) { Path srcRoot = BASE.resolve(id); for (String type : List.of("train", "val")) {