diff --git a/src/main/java/com/kamco/cd/training/model/dto/ModelTrainMngDto.java b/src/main/java/com/kamco/cd/training/model/dto/ModelTrainMngDto.java index 7b5a921..1c85da1 100644 --- a/src/main/java/com/kamco/cd/training/model/dto/ModelTrainMngDto.java +++ b/src/main/java/com/kamco/cd/training/model/dto/ModelTrainMngDto.java @@ -51,6 +51,8 @@ public class ModelTrainMngDto { private Long beforeModelId; private Integer bestEpoch; + private UUID beforeModelUuid; + public String getStatusName() { if (this.statusCd == null || this.statusCd.isBlank()) return null; try { diff --git a/src/main/java/com/kamco/cd/training/postgres/entity/ModelMasterEntity.java b/src/main/java/com/kamco/cd/training/postgres/entity/ModelMasterEntity.java index 825ccec..cd7ef41 100644 --- a/src/main/java/com/kamco/cd/training/postgres/entity/ModelMasterEntity.java +++ b/src/main/java/com/kamco/cd/training/postgres/entity/ModelMasterEntity.java @@ -136,6 +136,9 @@ public class ModelMasterEntity { @Column(name = "tmp_file_err_message", columnDefinition = "TEXT") private String tmpFileErrMessage; + @Column(name = "before_model_uuid") + private UUID beforeModelUuid; + public ModelTrainMngDto.Basic toDto() { return new ModelTrainMngDto.Basic( this.id, @@ -157,6 +160,7 @@ public class ModelMasterEntity { this.packingStrtDttm, this.packingEndDttm, this.beforeModelId, - this.bestEpoch); + this.bestEpoch, + this.beforeModelUuid); } }