lint fix

scaleoutsystems · Jan 12, 2024 · 764bd5f · 764bd5f
1 parent 3d5b5af
commit 764bd5f
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/fedn/fedn/utils/plugins/androidhelper.py b/fedn/fedn/utils/plugins/androidhelper.py
@@ -1,4 +1,3 @@
-import json
 import os
 import struct
 import tempfile
@@ -17,9 +16,7 @@ def __init__(self):
         super().__init__()
 
     # function to calculate an incremental weighted average of the weights
-    def increment_average(
-            model, model_next, num_examples, total_examples
-    ):
+    def increment_average(self, model, model_next, num_examples, total_examples):
         """Incremental weighted average of model weights.
 
         :param model: Current model weights.
@@ -74,7 +71,7 @@ def save(self, weights, path=None):
         if not path:
             path = self.get_tmp_path()
 
-        byte_array = struct.pack("f"*len(weights),*weights)
+        byte_array = struct.pack("f"*len(weights), *weights)
         with open(path, "wb") as file:
             file.write(byte_array)