diff --git a/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java b/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java index d3f11c7..66a1851 100644 --- a/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java +++ b/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java @@ -6,6 +6,7 @@ import com.kamco.cd.training.model.dto.ModelTrainMngDto; import com.kamco.cd.training.model.service.TmpDatasetService; import com.kamco.cd.training.postgres.core.ModelTrainJobCoreService; import com.kamco.cd.training.postgres.core.ModelTrainMngCoreService; +import com.kamco.cd.training.train.dto.ModelTrainJobDto; import com.kamco.cd.training.train.dto.ModelTrainJobQueuedEvent; import com.kamco.cd.training.train.dto.TrainRunRequest; import java.io.IOException; @@ -139,7 +140,7 @@ public class TrainJobService { throw new IllegalStateException("이미 진행중입니다."); } - var lastJob = + ModelTrainJobDto lastJob = modelTrainJobCoreService .findLatestByModelId(modelId) .orElseThrow(() -> new IllegalStateException("이전 실행 이력이 없습니다."));