Fix formatting

Signed-off-by: Agrawal, Kush <[email protected]>
securefederatedai · Feb 21, 2025 · 22f5288 · 22f5288
1 parent 6101894
commit 22f5288
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/openfl/component/collaborator/collaborator.py b/openfl/component/collaborator/collaborator.py
@@ -342,7 +342,7 @@ def do_task(self, task, round_number) -> dict:
         del local_output_tensor_dict
         del input_tensor_dict
         return metrics
-    
+
     def get_numpy_dict_for_tensorkeys(self, tensor_keys):
         """Get tensor dictionary for specified tensorkey set.
 

diff --git a/openfl/databases/tensor_db.py b/openfl/databases/tensor_db.py
@@ -131,7 +131,9 @@ def cache_tensor(self, tensor_key_dict: Dict[TensorKey, np.ndarray]) -> None:
                 )
                 entries_to_add.append(new_entry)
 
-            self.tensor_db = pd.concat([self.tensor_db, *entries_to_add], ignore_index=True, copy=True)
+            self.tensor_db = pd.concat(
+                [self.tensor_db, *entries_to_add], ignore_index=True, copy=True
+            )
 
             del old_tensor_db
             entries_to_add.clear()

diff --git a/openfl/federated/task/runner_pt.py b/openfl/federated/task/runner_pt.py
@@ -6,6 +6,7 @@
 
 from copy import deepcopy
 from typing import Iterator, Tuple
+
 import numpy as np
 import torch
 import torch.nn as nn