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 8d6a7fa..33b53fd 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 @@ -443,7 +443,7 @@ public class ModelTrainMngCoreService { .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId)); master.setStatusCd(TrainStatusType.STOPPED.getId()); - master.setStep2State(TrainStatusType.STOPPED.getId()); + master.setStep1State(TrainStatusType.STOPPED.getId()); master.setLastError(errorMessage); master.setUpdatedUid(userUtil.getId()); master.setUpdatedDttm(ZonedDateTime.now()); diff --git a/src/main/java/com/kamco/cd/training/train/service/JobRecoveryOnStartupService.java b/src/main/java/com/kamco/cd/training/train/service/JobRecoveryOnStartupService.java index 9be8ec3..f2510e5 100644 --- a/src/main/java/com/kamco/cd/training/train/service/JobRecoveryOnStartupService.java +++ b/src/main/java/com/kamco/cd/training/train/service/JobRecoveryOnStartupService.java @@ -157,6 +157,7 @@ public class JobRecoveryOnStartupService { modelTrainJobCoreService.markPaused( job.getId(), modelId, "AUTO_KILLED_ON_SERVER_RESTART"); + log.info("job = {}", job); markStepStopByJobType(job, "AUTO_KILLED_ON_SERVER_RESTART"); } catch (Exception e) {