Unique process models in process models discovery (lava-nc#277)

* Unique process models in process models discovery Signed-off-by: bamsumit <[email protected]>
monkin77 · Jul 22, 2022 · 2c35606 · 2c35606
1 parent 0658454
commit 2c35606
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/lava/magma/compiler/compiler_graphs.py b/src/lava/magma/compiler/compiler_graphs.py
@@ -851,7 +851,8 @@ def _find_proc_models(proc: AbstractProcess) \
                                     import_module(proc_cls_mod)
                                 class_ = getattr(proc_cls_mod,
                                                  proc_model.__name__)
-                                proc_models.append(class_)
+                                if class_ not in proc_models:
+                                    proc_models.append(class_)
                     except Exception:
                         warnings.warn(
                             f"Cannot import module '{module}' when searching "

diff --git a/src/lava/magma/core/run_configs.py b/src/lava/magma/core/run_configs.py
@@ -179,7 +179,6 @@ def select(self,
         -------
         Selected ProcessModel class
         """
-
         num_pm = len(proc_models)
         # Case 0: No ProcessModels exist:
         # ------------------------------