전이학습 등록 시 uuid 추가

This commit is contained in:
2026-06-23 16:13:53 +09:00
parent ad750f0f06
commit 0ab0fdc43e

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());