From e54ce4959bca1fdeda01701989f8debd9467d4bd Mon Sep 17 00:00:00 2001
From: 潘志宝 <979469083@qq.com>
Date: 星期五, 03 一月 2025 10:08:20 +0800
Subject: [PATCH] Merge branch 'master' of http://dlindusit.com:53929/r/iailab-plat

---
 iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mdk/sample/SampleFactory.java |   11 +++++++++--
 1 files changed, 9 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 587c0ca..46a0f1a 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,17 +27,21 @@
      * @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;
     }
 
-    public SampleDataConstructor createSampelData(String typeA) {
+    public SampleDataConstructor createSampleData(String typeA) {
         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