Fix issue 156 on numpy default casting (#172)

pxLi · Feb 3, 2022 · 889e8c7 · 889e8c7
1 parent 5916248
commit 889e8c7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nvflare/app_common/shareablegenerators/full_model_shareable_generator.py b/nvflare/app_common/shareablegenerators/full_model_shareable_generator.py
@@ -57,7 +57,7 @@ def shareable_to_learnable(self, shareable: Shareable, fl_ctx: FLContext) -> Mod
             if dxo.data is not None:
                 model_diff = dxo.data
                 for v_name, v_value in model_diff.items():
-                    weights[v_name] += v_value
+                    weights[v_name] = weights[v_name] + v_value
         elif dxo.data_kind == DataKind.WEIGHTS:
             weights = dxo.data
             if not weights: