From e88fba9645a57535d858ce48da8e9d9a3dc84adc Mon Sep 17 00:00:00 2001
From: dongyukun <1208714201@qq.com>
Date: 星期四, 02 一月 2025 16:48:01 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master'

---
 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