diff --git a/src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java b/src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java index ba34833c..4b017b63 100644 --- a/src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java +++ b/src/main/java/com/kamco/cd/kamcoback/postgres/core/InferenceResultCoreService.java @@ -304,9 +304,11 @@ public class InferenceResultCoreService { public InferenceBatchSheet getInferenceResultByStatus(String status) { MapSheetLearnEntity entity = - mapSheetLearnRepository - .getInferenceResultByStatus(status) - .orElseThrow(() -> new EntityNotFoundException(status)); + mapSheetLearnRepository.getInferenceResultByStatus(status).orElse(null); + + if (entity == null) { + return null; + } InferenceBatchSheet inferenceBatchSheet = new InferenceBatchSheet(); inferenceBatchSheet.setId(entity.getId());