From 2b47c5b0e504a653fe40ae8b6749e14e228a89bd Mon Sep 17 00:00:00 2001
From: dongyukun <1208714201@qq.com>
Date: 星期四, 02 一月 2025 13:46:57 +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