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:
@@ -120,6 +120,15 @@ public class ModelTrainMngCoreService {
|
|||||||
entity.setTrainType(addReq.getTrainType()); // 일반, 전이
|
entity.setTrainType(addReq.getTrainType()); // 일반, 전이
|
||||||
entity.setBeforeModelId(addReq.getBeforeModelId());
|
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.setStatusCd(TrainStatusType.READY.getId());
|
||||||
entity.setStep1State(TrainStatusType.READY.getId());
|
entity.setStep1State(TrainStatusType.READY.getId());
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user