hyperparam_with_modeltype

2026-02-12 19:16:24 +09:00
parent 5455da1e96 9e803661cd
commit 12f9de7367
1 changed files with 5 additions and 1 deletions
--- 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
@@ -39,7 +39,11 @@ public class ModelMngRepositoryImpl implements ModelMngRepositoryCustom {
    BooleanBuilder builder = new BooleanBuilder();

    if (req.getStatus() != null && !req.getStatus().isEmpty()) {
-      builder.and(modelMasterEntity.statusCd.eq(req.getStatus()));
+      builder.and(
+          modelMasterEntity
+              .step1State
+              .eq(req.getStatus())
+              .or(modelMasterEntity.step2State.eq(req.getStatus())));
    }

    if (req.getModelNo() != null && !req.getModelNo().isEmpty()) {