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 |    7 +++++--
 1 files changed, 5 insertions(+), 2 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 55ad888..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,11 +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 = predictSampleInfoConstructor;
+            sampleInfoConstructor = scheduleSampleInfoConstructor;
         }
         return sampleInfoConstructor;
     }

--
Gitblit v1.9.3