diff --git a/src/main/java/com/kamco/cd/training/postgres/repository/model/ModelMngRepositoryImpl.java b/src/main/java/com/kamco/cd/training/postgres/repository/model/ModelMngRepositoryImpl.java index 3ebbafc..642b6cc 100644 --- a/src/main/java/com/kamco/cd/training/postgres/repository/model/ModelMngRepositoryImpl.java +++ b/src/main/java/com/kamco/cd/training/postgres/repository/model/ModelMngRepositoryImpl.java @@ -45,6 +45,8 @@ public class ModelMngRepositoryImpl implements ModelMngRepositoryCustom { builder.and(modelMasterEntity.modelNo.eq(req.getModelNo())); } + builder.and(modelMasterEntity.delYn.isFalse()); + List content = queryFactory .selectFrom(modelMasterEntity) diff --git a/src/main/java/com/kamco/cd/training/train/dto/TrainRunRequest.java b/src/main/java/com/kamco/cd/training/train/dto/TrainRunRequest.java index e294ce7..5761131 100644 --- a/src/main/java/com/kamco/cd/training/train/dto/TrainRunRequest.java +++ b/src/main/java/com/kamco/cd/training/train/dto/TrainRunRequest.java @@ -15,7 +15,7 @@ public class TrainRunRequest { // ======================== // 기본 // ======================== - private UUID datasetFolder; + private String datasetFolder; private UUID outputFolder; private String inputSize; private String cropSize; @@ -84,10 +84,6 @@ public class TrainRunRequest { private UUID uuid; - public String getDatasetFolder() { - return String.valueOf(this.datasetFolder); - } - public String getOutputFolder() { return String.valueOf(this.outputFolder); }