update isMultiPartyMlJob

WeBankBlockchain · Oct 30, 2024 · f3a54eb · f3a54eb
1 parent decbe3d
commit f3a54eb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/wedpr-common/protocol/src/main/java/com/webank/wedpr/common/protocol/JobType.java b/wedpr-common/protocol/src/main/java/com/webank/wedpr/common/protocol/JobType.java
@@ -71,7 +71,9 @@ public static Boolean isMultiPartyMlJob(String jobType) {
         return jobType.compareToIgnoreCase(XGB_TRAIN.getType()) == 0
                 || jobType.compareToIgnoreCase(XGB_PREDICT.getType()) == 0
                 || jobType.compareToIgnoreCase(LR_TRAIN.getType()) == 0
-                || jobType.compareToIgnoreCase(LR_PREDICT.getType()) == 0;
+                || jobType.compareToIgnoreCase(LR_PREDICT.getType()) == 0
+                || jobType.compareToIgnoreCase(MLPreprocessing.getType()) == 0
+                || jobType.compareToIgnoreCase(FeatureEngineer.getType()) == 0;
     }
 
     public static Boolean isPSIJob(String jobType) {