Merge pull request '전이학습 등록 시 uuid 추가' (#203) from feat/training_260623 into develop

Reviewed-on: #203
This commit was merged in pull request #203.
This commit is contained in:
2026-06-23 16:14:46 +09:00

View File

@@ -120,6 +120,15 @@ public class ModelTrainMngCoreService {
entity.setTrainType(addReq.getTrainType()); // 일반, 전이
entity.setBeforeModelId(addReq.getBeforeModelId());
// 전이학습일 때 beforeModelUuid 추가하기
if (addReq.getBeforeModelId() != null) {
ModelMasterEntity beforeModel =
modelMngRepository
.findById(addReq.getBeforeModelId())
.orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
entity.setBeforeModelUuid(beforeModel.getUuid());
}
entity.setStatusCd(TrainStatusType.READY.getId());
entity.setStep1State(TrainStatusType.READY.getId());