Fix codacy issues in callback_fx.py

lava-nc · Jun 28, 2023 · e189d8a · e189d8a
1 parent 2b069fe
commit e189d8a
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/src/lava/magma/core/callback_fx.py b/src/lava/magma/core/callback_fx.py
@@ -61,16 +61,26 @@ class IterableCallBack(NxSdkCallbackFx):
     as pre and post run."""
 
     def __init__(self,
-                 pre_run_fxs: Iterable = [],
-                 post_run_fxs: Iterable = []) -> None:
+                 pre_run_fxs: Iterable = None,
+                 post_run_fxs: Iterable = None) -> None:
         super().__init__()
+        if pre_run_fxs is None:
+            pre_run_fxs = []
+        if post_run_fxs is None:
+            post_run_fxs = []
         self.pre_run_fxs = pre_run_fxs
         self.post_run_fxs = post_run_fxs
 
-    def pre_run_callback(self, board: NxBoard, **_) -> None:
+    def pre_run_callback(self,
+                         board: NxBoard,
+                         _var_id_to_var_model_map: dict = None
+                        ) -> None:
         for fx in self.pre_run_fxs:
             fx(board)
 
-    def post_run_callback(self, board: NxBoard, **_) -> None:
+    def post_run_callback(self,
+                         board: NxBoard,
+                         _var_id_to_var_model_map: dict = None
+                        ) -> None:
         for fx in self.post_run_fxs:
             fx(board)