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 4d1f621..345218e 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,7 @@ 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.TmpDatasetService; +import com.kamco.cd.training.train.service.TrainJobService; import java.util.List; import java.util.UUID; import lombok.RequiredArgsConstructor; @@ -30,7 +30,7 @@ public class ModelTrainMngService { private final ModelTrainMngCoreService modelTrainMngCoreService; private final HyperParamCoreService hyperParamCoreService; - private final TmpDatasetService tmpDatasetService; + private final TrainJobService trainJobService; /** * 모델학습 조회 @@ -93,6 +93,8 @@ public class ModelTrainMngService { // 모델 config 저장 modelTrainMngCoreService.saveModelConfig(modelId, req.getModelConfig()); + // 임시파일 생성 + trainJobService.createTmpFile(modelUuid); return modelUuid; } diff --git a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java index e8f99bd..2972b05 100644 --- a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java +++ b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java @@ -159,7 +159,6 @@ public class ModelTrainMngCoreService { * @param modelId * @param req */ - @Transactional public void updateModelMaster(Long modelId, ModelTrainMngDto.UpdateReq req) { ModelMasterEntity entity = modelMngRepository diff --git a/src/main/java/com/kamco/cd/training/train/TrainApiController.java b/src/main/java/com/kamco/cd/training/train/TrainApiController.java index 149ebdf..941278b 100644 --- a/src/main/java/com/kamco/cd/training/train/TrainApiController.java +++ b/src/main/java/com/kamco/cd/training/train/TrainApiController.java @@ -49,7 +49,6 @@ public class TrainApiController { @PathVariable UUID uuid) { Long modelId = trainJobService.getModelIdByUuid(uuid); - trainJobService.createTmpFile(uuid); trainJobService.enqueue(modelId); return ApiResponseDto.ok("ok"); 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 0500a31..505e9ed 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 @@ -219,6 +219,7 @@ public class TrainJobService { } } + @Transactional public UUID createTmpFile(UUID modelUuid) { UUID tmpUuid = UUID.randomUUID(); String raw = tmpUuid.toString().toUpperCase().replace("-", "");