From a23e66f179586eeffc55d116f5a3bb53eab9cb09 Mon Sep 17 00:00:00 2001
From: dongyukun <1208714201@qq.com>
Date: 星期五, 03 一月 2025 14:00:17 +0800
Subject: [PATCH] 调度方案批量启禁
---
iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/sample/SampleFactory.java | 9 ++++++++-
1 files changed, 8 insertions(+), 1 deletions(-)
diff --git a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/sample/SampleFactory.java b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/sample/SampleFactory.java
index 587c0ca..194b6a3 100644
--- a/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/sample/SampleFactory.java
+++ b/iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/sample/SampleFactory.java
@@ -14,6 +14,9 @@
private PredictSampleInfoConstructor predictSampleInfoConstructor;
@Autowired
+ private ScheduleSampleInfoConstructor scheduleSampleInfoConstructor;
+
+ @Autowired
private PredictSampleDataConstructor predictSampleDataConstructor;
/**
@@ -24,9 +27,11 @@
* @return
*/
public SampleInfoConstructor createSampleInfo(String typeA, String modelId){
- PredictSampleInfoConstructor sampleInfoConstructor = null;
+ SampleInfoConstructor sampleInfoConstructor = null;
if (typeA.compareTo(TypeA.Predict.name()) == 0) {
sampleInfoConstructor = predictSampleInfoConstructor;
+ } else if (typeA.compareTo(TypeA.Schedule.name()) == 0) {
+ sampleInfoConstructor = scheduleSampleInfoConstructor;
}
return sampleInfoConstructor;
}
@@ -35,6 +40,8 @@
SampleDataConstructor sampleDataConstructor = null;
if (typeA.compareTo(TypeA.Predict.name()) == 0) {
sampleDataConstructor = predictSampleDataConstructor;
+ } else if (typeA.compareTo(TypeA.Schedule.name()) == 0) {
+ sampleDataConstructor = predictSampleDataConstructor;
}
return sampleDataConstructor;
}
--
Gitblit v1.9.3