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 6b54aa9..e1fe664 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 @@ -419,6 +419,11 @@ public class ModelTrainMngCoreService { .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId)); master.setStatusCd(TrainStatusType.STOPPED.getId()); + if (master.getStep2StrtDttm() != null) { + master.setStep2State(TrainStatusType.STOPPED.getId()); + } else { + master.setStep1State(TrainStatusType.STOPPED.getId()); + } } /** 완료 처리(옵션) - Worker가 성공 시 호출 */