Merge branch 'feat/training_260202' of https://kamco.git.gs.dabeeo.com/MVPTeam/kamco-train-api into feat/training_260202

This commit is contained in:
2026-02-13 12:23:12 +09:00
4 changed files with 5 additions and 4 deletions

View File

@@ -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;
}

View File

@@ -159,7 +159,6 @@ public class ModelTrainMngCoreService {
* @param modelId
* @param req
*/
@Transactional
public void updateModelMaster(Long modelId, ModelTrainMngDto.UpdateReq req) {
ModelMasterEntity entity =
modelMngRepository

View File

@@ -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");

View File

@@ -219,6 +219,7 @@ public class TrainJobService {
}
}
@Transactional
public UUID createTmpFile(UUID modelUuid) {
UUID tmpUuid = UUID.randomUUID();
String raw = tmpUuid.toString().toUpperCase().replace("-", "");