Iterator callback fx signature fix (#743)

* update refport unittest to always wait when it writes to port for consistent behavior Signed-off-by: bamsumit <[email protected]> * Removed pyproject changes Signed-off-by: bamsumit <[email protected]> * Fix to convolution tests. Fixed imcompatible mnist_pretrained for old python versions. Signed-off-by: bamsumit <[email protected]> * Missing moudle parent fix Signed-off-by: bamsumit <[email protected]> * Added ConvVarModel * Added iterable callback function Signed-off-by: bamsumit <[email protected]> * Fix codacy issues in callback_fx.py * Fix linting in callback_fx.py * Fix codacy sig issue in callback_fx.py --------- Signed-off-by: bamsumit <[email protected]> Co-authored-by: Joyesh Mishra <[email protected]> Co-authored-by: Marcus G K Williams <[email protected]>
lava-nc · Oct 24, 2023 · e76d3f6 · e76d3f6
1 parent 617332f
commit e76d3f6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/lava/magma/core/callback_fx.py b/src/lava/magma/core/callback_fx.py
@@ -73,14 +73,14 @@ def __init__(self,
 
     def pre_run_callback(self,
                          board: NxBoard = None,
-                         _var_id_to_var_model_map: dict = None
+                         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,
-                          _var_id_to_var_model_map: dict = None
+                          var_id_to_var_model_map: dict = None
                           ) -> None:
         for fx in self.post_run_fxs:
             fx(board)