update hetero-nn

Signed-off-by: weijingchen <talkingwallace@sohu.com> Signed-off-by: cwj <talkingwallace@sohu.com>
FederatedAI · Nov 27, 2023 · c9f5f82 · c9f5f82
1 parent 26f96e3
commit c9f5f82
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/fate/ml/aggregator/base.py b/python/fate/ml/aggregator/base.py
@@ -149,6 +149,9 @@ def _process_model(self, model):
                 to_agg.append(self._convert_type(p, self.float_p))
             agg_list = to_agg
 
+        else:
+            return None
+
         return agg_list
 
     def _recover_model(self, model, agg_model):